Pinned Loading

1. JaxMARL JaxMARL Public

   Multi-Agent Reinforcement Learning with JAX

   Python 646 126

2. Kinetix Kinetix Public

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

   Python 213 10

3. jafar jafar Public

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

   Python 83 18

4. jaxirl jaxirl Public

   Contains JAX implementation of algorithms for inverse reinforcement learning

   Python 72 3