#90 · Primary category: Deep Learning Frameworks
rl-baselines3-zoo
A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.
Project last updated:08/24/26
GitHub Stars
2.9K
Forks
605
Contributors
40
License
MIT
Why we included this project
Anyone doing serious work with Stable Baselines3 will probably end up using this project. It bundles the practical parts of the RL workflow: training scripts, evaluation, video recording, and plotting, plus tuned hyperparameters for common environments and algorithms. The hyperparameters live in YAML files you can copy and adjust as a starting point, and the included pre-trained agents let you watch a well-configured policy behave before spending compute on your own training runs. That makes it handy for benchmarking algorithms and reproducing results, since you get sane defaults and reproducible configs in one place. It is also a decent way to learn how hyperparameter choices translate into RL performance.
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.