Stable Baselines3

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:


RL Baselines3 Zoo (collection of pre-trained agents):

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

Antonin Raffin
Research Engineer in Robotics and Machine Learning

Robots. Machine Learning. Blues Dance.