instadeepai / matrax
A collection of matrix games in JAX
β9Updated 2 months ago
Alternatives and similar repositories for matrax:
Users that are interested in matrax are comparing it to the libraries listed below
- πͺ The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAXβ55Updated last year
- A categorised list of Multi-Agent Reinforcemnt Learning (MARL) papersβ48Updated 2 years ago
- Simple JAX Graphics Library.β29Updated 3 months ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorchβ47Updated last year
- β18Updated 2 weeks ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)β9Updated last year
- A tool for aggregating and plotting MARL experiment data.β71Updated 3 weeks ago
- β70Updated 5 months ago
- An Open-Ended Agentic Simulatorβ39Updated 6 months ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancyβ17Updated 3 months ago
- β73Updated 3 months ago
- Highly scalable 2D JAX physics engine.β44Updated 2 weeks ago
- Accelerated replay buffers in JAXβ41Updated 2 years ago
- The Starcraft Multi-Agent challenge liteβ41Updated 5 months ago
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023β19Updated last year
- β41Updated last year
- POPGym Library in JAXβ11Updated 10 months ago
- β20Updated 7 months ago
- Baselines for gymnax π€β63Updated last year
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"β64Updated 8 months ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselinβ¦β47Updated last week
- Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"β19Updated 2 months ago
- Learning diverse options through the Laplacian representation.β23Updated last year
- Scalable Opponent Shaping Experiments in JAXβ24Updated 10 months ago
- Code of the paper: Debiasing Meta-Gradient Reinforcement Learning by Learning the Outer Value Functionβ13Updated 2 years ago
- IMP-MARL: a Suite of Environments for Large-scale Infrastructure Management Planning via MARLβ36Updated 5 months ago
- β47Updated 2 years ago
- JAX implementations of various deep reinforcement learning algorithms.β20Updated 2 weeks ago
- HPO and Architecture Benchmarking for RL: Dynamically, Reactive and Efficientβ17Updated this week
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learningβ14Updated 2 years ago