Jax version of Stable Baselines, implementations of reinforcement learning algorithms.
Project description
Stable Baselines Jax (SB3 + JAX = SBX)
See https://github.com/araffin/sbx
Example
from sbx import TQC, DroQ, SAC
model = TQC("MlpPolicy", "Pendulum-v1", verbose=1)
model.learn(total_timesteps=10_000)
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
sbx-rl-0.1.0.post1.tar.gz
(13.8 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file sbx-rl-0.1.0.post1.tar.gz.
File metadata
- Download URL: sbx-rl-0.1.0.post1.tar.gz
- Upload date:
- Size: 13.8 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.7.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
d219c77014b41694496ed816f5c44ef5a730922ce90b50524c2d028edba50c8c
|
|
| MD5 |
2e1d49ba1d98ed502739ad57c39ad04b
|
|
| BLAKE2b-256 |
29955d3228ac88a5f365fad687c11405fb874e157e74ec5148640ca431dabf9d
|
File details
Details for the file sbx_rl-0.1.0.post1-py3-none-any.whl.
File metadata
- Download URL: sbx_rl-0.1.0.post1-py3-none-any.whl
- Upload date:
- Size: 21.3 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.7.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
086e4dd279b84f93986b3e2c5bfba53c8206c758b57362366165e7945d7ba7a6
|
|
| MD5 |
da5548e8a04ec4313a1039693cf7c191
|
|
| BLAKE2b-256 |
f2fffbc3740eb7407de741e3d913dcfd79143a877ad5891c01306993fc21ce07
|
