A reinforcement library designed for pytorch
machin
Machin is a reinforcement library designed for pytorch.
Reinforcement learning library(framework) designed for PyTorch, implements DQN, DDPG, A2C, PPO, SAC, MADDPG, A3C, APEX, IMPALA …
Features
1. Automatic
Starting from version 0.4.0, Machin now supports automatic config generation, you can get a configuration
through:
python -m machin.auto generate --algo DQN --env openai_gym --output config.json
And automatically launch the experiment with pytorch lightning:
python -m machin.auto launch --config config.json
2. Readable
Compared to other reinforcement learning libraries such as the famous rlpyt, ray, and baselines. Machin tries to just provide a simple, clear implementation of RL algorithms.
All algorithms in Machin are designed with minimial abstractions and have very detailed documents, as well as various helpful tutorials.
3. Reusable
Machin takes