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 files for sbx-rl 0.2.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| sbx-rl-0.2.0.tar.gz | 13.8 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| sbx_rl-0.2.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 34.6 kB
Release files / sbx-rl-0.2.0.tar.gz
| Download URL | sbx-rl-0.2.0.tar.gz |
|---|---|
| Size | 13.8 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
7c0d4b72bbf654279bf0002503a230a4a6368f48fcfdc2a82a05f7ed263c0f25
|
|
BLAKE2b-256 checksum How to use checksums |
9909a981832be28de4c946ba2b24a6399d75c0be03beb5a887d738ab210ccf9e
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.1 CPython/3.7.12
|
Release files / sbx_rl-0.2.0-py3-none-any.whl
| Download URL | sbx_rl-0.2.0-py3-none-any.whl |
|---|---|
| Size | 20.8 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
e1e30e4f7513f59d2f3b8413903174448a9dccb47764e69bb916fafb5a377049
|
|
BLAKE2b-256 checksum How to use checksums |
11e3f8c77020f26059d5851617f03c76194d3868dd832cf2cadcbdfc15e3930f
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.1 CPython/3.7.12
|
