/jax-baseline

Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselines.

Primary LanguagePythonMIT LicenseMIT

Stargazers