Jax version of Stable Baselines, implementations of reinforcement learning algorithms.

Navigation

Verified details

These details have been verified by PyPI
Maintainers
Avatar for araffin from gravatar.com araffin

Unverified details

These details have not been verified by PyPI
Project links
Meta
  • License: MIT
  • Author: Antonin Raffin
  • Tags reinforcement-learning-algorithms, reinforcement-learning, machine-learning, gym, openai, stable, baselines, toolbox, python, data-science
  • Requires: Python >=3.7
Classifiers

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

Verified details

These details have been verified by PyPI
Maintainers
Avatar for araffin from gravatar.com araffin

Unverified details

These details have not been verified by PyPI
Project links
Meta
  • License: MIT
  • Author: Antonin Raffin
  • Tags reinforcement-learning-algorithms, reinforcement-learning, machine-learning, gym, openai, stable, baselines, toolbox, python, data-science
  • Requires: Python >=3.7
Classifiers

Release history Release notifications | RSS feed

0.18.0

0.17.0

0.15.0

0.13.0

0.12.0

0.11.0

0.10.0

0.9.1

0.9.0

0.8.0

0.7.0

0.6.0

0.5.0

0.3.0

This version

0.2.0

0.1.0.post1

0.0.3

0.0.2

0.0.1

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 hashes)

Uploaded Source

Built Distribution

sbx_rl-0.2.0-py3-none-any.whl (20.8 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page