Stable Baselines Jax (SB3 + JAX = SBX)
See https://github.com/araffin/sbx
Proof of concept version of Stable-Baselines3 in Jax.
Implemented algorithms:
- Soft Actor-Critic (SAC) and SAC-N
- Truncated Quantile Critics (TQC)
- Dropout Q-Functions for Doubly Efficient Reinforcement Learning (DroQ)
- Proximal Policy Optimization (PPO)
- Deep Q Network (DQN)
- Twin Delayed DDPG (TD3)
- Deep Deterministic Policy Gradient (DDPG)
Example
from sbx import TQC, DroQ, SAC, DQN, PPO, TD3, DDPG
model = TQC("MlpPolicy", "Pendulum-v1", verbose=1)
model.learn(total_timesteps=10_000, progress_bar=True)
Release files for sbx-rl 0.8.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.8.0.tar.gz | 29.3 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| sbx_rl-0.8.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 73.4 kB
Release files / sbx-rl-0.8.0.tar.gz
| Download URL | sbx-rl-0.8.0.tar.gz |
|---|---|
| Size | 29.3 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
836dbb53001d6ca0c075b4b033802800cde57aed88b0364f5908e441369a083b
|
|
BLAKE2b-256 checksum How to use checksums |
a9721798582e870f8257bcf04d49dd1d164ceb2e179752248a137119ce4853a2
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.2 CPython/3.9.13
|
Release files / sbx_rl-0.8.0-py3-none-any.whl
| Download URL | sbx_rl-0.8.0-py3-none-any.whl |
|---|---|
| Size | 44.1 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
2b435ec2d22e0d65914c7724c9acb1dffe37c7f496b9012c39d8340582fd0c3f
|
|
BLAKE2b-256 checksum How to use checksums |
f42a1cf3c19d31e7d9795258be738be25262f28cff2d620503978d04ea7bd1d7
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.2 CPython/3.9.13
|
