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)
Project details
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 hashes)
Built Distribution
Close
Hashes for sbx_rl-0.1.0.post1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 086e4dd279b84f93986b3e2c5bfba53c8206c758b57362366165e7945d7ba7a6 |
|
MD5 | da5548e8a04ec4313a1039693cf7c191 |
|
BLAKE2b-256 | f2fffbc3740eb7407de741e3d913dcfd79143a877ad5891c01306993fc21ce07 |