a modular reinforcement learning library with JAX agents
machine-learning reinforcement-learning deep-reinforcement-learning gym exploration flax sac jax td3 brax tqc sdqn
-
Updated
Mar 3, 2025 - Python