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
- πͺ The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAXβ57Updated last year
- β18Updated last month
- A tool for aggregating and plotting MARL experiment data.β75Updated last month
- A categorised list of Multi-Agent Reinforcemnt Learning (MARL) papersβ50Updated 2 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorchβ48Updated last year
- β74Updated 6 months ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)β9Updated last year
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancyβ17Updated 4 months ago
- β41Updated last year
- The Starcraft Multi-Agent challenge liteβ42Updated 6 months ago
- β73Updated 4 months ago
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023β19Updated last year
- Accelerated replay buffers in JAXβ41Updated 2 years ago
- β47Updated 2 years ago
- Code for magnetic mirror descent.β15Updated last year
- POPGym Library in JAXβ11Updated 11 months ago
- Baselines for gymnax π€β66Updated last year
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"β67Updated 9 months ago
- An Open-Ended Agentic Simulatorβ45Updated 7 months ago
- Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"β21Updated 3 months ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselinβ¦β48Updated last month
- A neural network library written in jaxβ11Updated last month
- Conservative Q learning in Jaxβ53Updated 2 years ago
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learningβ14Updated 2 years ago
- Vectorization techniques for fast population-based training.β55Updated 2 years ago
- β20Updated 9 months ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"β100Updated 2 years ago
- Various reinforcement learning algorithms written in Jax + Flaxβ24Updated last year
- Official implementation of the Ξ΄-model presented in the ICML 2024 paper "A Distributional Analogue to the Successor Representation".β20Updated 4 months ago