Reinforcement Learning and Generative Models

Hey everyone.
So for the past 6-7 months, I have been working on maintaining a single library for all pytorch reinforcement learning algorithms (as well as generative models). Something similar to keras-rl. Do check it out.
It is still under active development. Feel free to contribute to the repository if you have any particular algorithm in mind or you want some changes to existing ones (such as adding parallelization of the policy gradient algorithms).

Pytorch-RL

3 Likes

Are there any plans to implement imitation learning stuff?

1 Like