Last released Jan 12, 2025
Reinforcement learning in pure JAX.
Last released May 30, 2022
A flexible trainer interface for Jax and Haiku.
Supported by