#28 · Primary category: Deep Learning Frameworks
stable-baselines3
PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.
Project last updated:08/17/26
GitHub Stars
13.7K
Forks
2.2K
Contributors
174
License
MIT
Why we included this project
Most RL teams pick Stable Baselines3 when they want tested, consistent PyTorch implementations of the standard algorithms without writing training loops from scratch. PPO, DQN, SAC, TD3, and A2C all share one API, so swapping methods and comparing results is straightforward. Every algorithm is benchmarked against reference codebases and covered by unit tests, and the library handles custom policies, dictionary observation spaces, callbacks, and TensorBoard logging out of the box. That makes it a practical starting point for research experiments and quick prototypes in robotics, control, or game-like decision problems. Just note that it assumes you already know the basics of reinforcement learning, so it is not a beginner's introduction to the field.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
tensorflow
An Open Source Machine Learning Framework for Everyone
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras
Deep Learning for humans
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.