instadeepai / matraxLinks
A collection of matrix games in JAX
β12Updated 10 months ago
Alternatives and similar repositories for matrax
Users that are interested in matrax are comparing it to the libraries listed below
Sorting:
- Code for Discovered Policy Optimisation (NeurIPS 2022)β12Updated 2 years ago
- πͺ The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAXβ59Updated last year
- β83Updated last month
- A categorised list of Multi-Agent Reinforcemnt Learning (MARL) papersβ55Updated 2 years ago
- Accelerated minigrid environments with JAXβ150Updated last month
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancyβ20Updated 11 months ago
- Drop-in environment replacements that make your RL algorithm train faster.β21Updated last year
- An Open-Ended Agentic Simulatorβ52Updated last year
- Accelerated replay buffers in JAXβ43Updated 3 years ago
- β‘ Flashbax: Accelerated Replay Buffers in JAXβ253Updated 2 weeks ago
- Simple single-file baselines for Q-Learning in pure-GPU settingβ184Updated 6 months ago
- Baselines for gymnax π€β72Updated 2 years ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLβ115Updated last year
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]β110Updated last year
- Contains JAX implementation of algorithms for inverse reinforcement learningβ72Updated last year
- β84Updated 11 months ago
- Unified Implementations of Offline Reinforcement Learning Algorithmsβ111Updated 5 months ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!β236Updated 4 months ago
- Challenging Memory-based Deep Reinforcement Learning Agentsβ103Updated 11 months ago
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learningβ17Updated 2 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorchβ52Updated 2 years ago
- A tool for aggregating and plotting MARL experiment data.β78Updated 8 months ago
- Simple JAX Graphics Library.β36Updated 11 months ago
- Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"β28Updated 5 months ago
- JAX implementation of RL algorithms and vectorized environmentsβ49Updated last year
- POPGym Library in JAXβ11Updated last year
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselinβ¦β58Updated 2 weeks ago
- JAX implementations of core Deep RL algorithmsβ82Updated 3 years ago
- Learning diverse options through the Laplacian representation.β23Updated last year
- Vectorization techniques for fast population-based training.β56Updated 3 years ago