Pinned Loading

1. JaxMARL JaxMARL Public

   Multi-Agent Reinforcement Learning with JAX

   Python 763 141

2. Kinetix Kinetix Public

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

   Python 233 10

3. jafar jafar Public

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

   Python 102 23

4. jaxirl jaxirl Public

   Contains JAX implementation of algorithms for inverse reinforcement learning

   Python 74 4