instadeepai / matrax
A collection of matrix games in JAX
ā9Updated 3 months ago
Alternatives and similar repositories for matrax:
Users that are interested in matrax are comparing it to the libraries listed below
- ā18Updated last month
- šŖ The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAXā57Updated last year
- A categorised list of Multi-Agent Reinforcemnt Learning (MARL) papersā50Updated 2 years ago
- A tool for aggregating and plotting MARL experiment data.ā75Updated last month
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorchā48Updated last year
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancyā17Updated 4 months ago
- Accelerated replay buffers in JAXā41Updated 2 years ago
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023ā19Updated last year
- ā74Updated 6 months ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)ā9Updated last year
- ā41Updated last year
- Baselines for gymnax š¤ā66Updated last year
- An Open-Ended Agentic Simulatorā45Updated 7 months ago
- POPGym Library in JAXā11Updated 11 months ago
- Fast and flexible multi-agent gridworld reinforcement learning environments.ā40Updated 2 months ago
- Repository for the PGA-MAP-Elites algorithm. PGA-MAP-Elites was developed to efficiently scale MAP-Elites to large genotypes and noisy dā¦ā36Updated 3 years ago
- The Starcraft Multi-Agent challenge liteā42Updated 6 months ago
- ā73Updated 4 months ago
- Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"ā21Updated 3 months ago
- Conservative Q learning in Jaxā53Updated 2 years ago
- Scalable Opponent Shaping Experiments in JAXā24Updated 11 months ago
- Official implementation of the Ī“-model presented in the ICML 2024 paper "A Distributional Analogue to the Successor Representation".ā20Updated 4 months ago
- Challenging Memory-based Deep Reinforcement Learning Agentsā93Updated 4 months ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"ā67Updated 9 months ago
- Simple JAX Graphics Library.ā34Updated 4 months ago
- Learning diverse options through the Laplacian representation.ā23Updated last year
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learningā14Updated 2 years ago
- ā47Updated 2 years ago
- Skeleton for scalable and flexible Jax RL implementationsā73Updated last year
- Contains JAX implementation of algorithms for inverse reinforcement learningā69Updated 6 months ago