S-RL Toolbox: Reinforcement Learning (RL) and State Representation Learning (SRL) Toolbox for Robotics.

Github repository: https://github.com/araffin/robotics-rl-srl

Documentation: https://s-rl-toolbox.readthedocs.io

Paper: https://arxiv.org/abs/1809.09369

Associated Stable Baselines Medium post: https://towardsdatascience.com/stable-baselines-a-fork-of-openai-baselines-reinforcement-learning-made-easy-df87c4b2fc82?source=user_profile---------2------------------

Main Features

• 10 RL algorithms (Stable Baselines included)
• logging / plotting / visdom integration / replay trained agent
• hyperparameter search (hyperband, hyperopt)
• integration with State Representation Learning (SRL) methods (for feature extraction)
• visualisation tools (explore latent space, display action proba, live plot in the state space, …)
• robotics environments to compare SRL methods
• easy install using anaconda env or Docker images (CPU/GPU)