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.2.0.tar.gz
(13.8 kB
view details)
Built Distribution
sbx_rl-0.2.0-py3-none-any.whl
(20.8 kB
view details)
File details
Details for the file sbx-rl-0.2.0.tar.gz.
File metadata
- Download URL: sbx-rl-0.2.0.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 |
7c0d4b72bbf654279bf0002503a230a4a6368f48fcfdc2a82a05f7ed263c0f25
|
|
| MD5 |
a894992e8802b7fcba2f1869b0d99a23
|
|
| BLAKE2b-256 |
9909a981832be28de4c946ba2b24a6399d75c0be03beb5a887d738ab210ccf9e
|
File details
Details for the file sbx_rl-0.2.0-py3-none-any.whl.
File metadata
- Download URL: sbx_rl-0.2.0-py3-none-any.whl
- Upload date:
- Size: 20.8 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 |
e1e30e4f7513f59d2f3b8413903174448a9dccb47764e69bb916fafb5a377049
|
|
| MD5 |
9510d7c7aef1bb0917060a0661b7b310
|
|
| BLAKE2b-256 |
11e3f8c77020f26059d5851617f03c76194d3868dd832cf2cadcbdfc15e3930f
|
