Last released Dec 29, 2023
Reinforcement learning in pure JAX.
Last released May 30, 2022
A flexible trainer interface for Jax and Haiku.
Supported by