saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-05-14 20:58:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[ci skip] Update README with SBX

Browse source
This commit is contained in:
Antonin Raffin 2023-06-23 12:38:31 +02:00
parent cc103ff725
commit 472ff8edb8
No known key found for this signature in database
GPG key ID: B8B48F65CAD6232C
2 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
README.md
View file

@ -83,6 +83,12 @@ This allows SB3 to maintain a stable and compact core, while still providing the
Documentation is available online: [https://sb3-contrib.readthedocs.io/](https://sb3-contrib.readthedocs.io/)
## Stable-Baselines Jax (SBX)
[Stable Baselines Jax (SBX)](https://github.com/araffin/sbx) is a proof of concept version of Stable-Baselines3 in Jax.
It provides a minimal number of features compared to SB3 but can be much faster (up to 20x times!): https://twitter.com/araffin2/status/1590714558628253698
## Installation

1
setup.py
View file

@ -162,6 +162,7 @@ setup(
"Changelog": "https://stable-baselines3.readthedocs.io/en/master/misc/changelog.html",
"SB3-Contrib": "https://github.com/Stable-Baselines-Team/stable-baselines3-contrib",
"RL-Zoo": "https://github.com/DLR-RM/rl-baselines3-zoo",
"SBX": "https://github.com/araffin/sbx",
},
classifiers=[
"Programming Language :: Python :: 3",