Educational implementation of RL for LLMs in Pytorch - View it on GitHub