Pinned Loading

1. JaxMARL JaxMARL Public

   Multi-Agent Reinforcement Learning with JAX

   Python 707 134

2. Kinetix Kinetix Public

   Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.

   Python 226 9

3. jafar jafar Public

   JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"

   Python 97 21

4. jaxirl jaxirl Public

   Contains JAX implementation of algorithms for inverse reinforcement learning

   Python 74 4