Pinned Loading

1. JaxMARL JaxMARL Public

   Multi-Agent Reinforcement Learning with JAX

   Python 835 157

2. Kinetix Kinetix Public

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

   Python 262 12

3. jafar jafar Public

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

   Python 107 25

4. jaxirl jaxirl Public

   Contains JAX implementation of algorithms for inverse reinforcement learning

   Python 81 6