Pinned Loading

1. JaxMARL JaxMARL Public

   Multi-Agent Reinforcement Learning with JAX

   Python 465 89

2. Kinetix Kinetix Public

   Reinforcement learning on general 2D physics environments in JAX.

   Python 117 3

3. jaxirl jaxirl Public

   Contains JAX implementation of algorithms for inverse reinforcement learning

   Python 65 1

4. jafar jafar Public

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

   Python 48 4

5. JaxGL JaxGL Public

   Simple JAX Graphics Library.

   Python 27

6. popjym popjym Public

   POPGym Library in JAX

   Python 11