Welcome to Stable Baselines3 docs! - RL Baselines Made Easy

Stable Baselines3 is a set of improved implementations of reinforcement learning algorithms in PyTorch. It is the next major version of Stable Baselines.

Github repository: https://github.com/DLR-RM/stable-baselines3

RL Baselines3 Zoo (collection of pre-trained agents): https://github.com/DLR-RM/rl-baselines3-zoo

RL Baselines3 Zoo also offers a simple interface to train, evaluate agents and do hyperparameter tuning.

Main Features

  • Unified structure for all algorithms

  • PEP8 compliant (unified code style)

  • Documented functions and classes

  • Tests, high code coverage and type hints

  • Clean code

  • Tensorboard support

RL Algorithms

Citing Stable Baselines3

To cite this project in publications:

  author = {Raffin, Antonin and Hill, Ashley and Ernestus, Maximilian and Gleave, Adam and Kanervisto, Anssi and Dormann, Noah},
  title = {Stable Baselines3},
  year = {2019},
  publisher = {GitHub},
  journal = {GitHub repository},
  howpublished = {\url{https://github.com/DLR-RM/stable-baselines3}},

Indices and tables