S-RL Toolbox

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

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)
Antonin Raffin
Research Engineer in Robotics and Machine Learning

Robots. Machine Learning. Blues Dance.