Pinned

1. purejaxrl purejaxrl Public

   Really Fast End-to-End Jax RL Implementations

   Python 564 46

2. FLAIROx/JaxMARL FLAIROx/JaxMARL Public

   Multi-Agent Reinforcement Learning with JAX

   Python 328 50

3. popjaxrl popjaxrl Public

   Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]

   Python 58 8

4. model-free-opponent-shaping model-free-opponent-shaping Public

   Code for Model-Free Opponent Shaping (ICML 2022)

   Python 17 9

5. discovered-policy-optimisation discovered-policy-optimisation Public

   Code for Discovered Policy Optimisation (NeurIPS 2022)

   Python 5

6. adversarial-cheap-talk adversarial-cheap-talk Public

   Code for Adversarial Cheap Talk (ICML 2023)

   Python 3 1