kinalmehta / marl-jaxView external linksLinks
JAX library for MARL research
☆87Oct 28, 2023Updated 2 years ago
Alternatives and similar repositories for marl-jax
Users that are interested in marl-jax are comparing it to the libraries listed below
Sorting:
- Official code for "A General Learning Framework for Open Ad Hoc Teamwork Using Graph-based Policy Learning"☆17Mar 1, 2023Updated 2 years ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆74Aug 18, 2024Updated last year
- Modular Single-file Reinfocement Learning Algorithms Library☆38May 16, 2023Updated 2 years ago
- [NeurIPS 2022] Open source code for reusing prior computational work in RL.☆100Jul 5, 2023Updated 2 years ago
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)☆19May 10, 2024Updated last year
- ☆18Aug 30, 2020Updated 5 years ago
- Codebase for BRDiv: Diverse teammate generation for ad hoc teamwork☆13May 2, 2024Updated last year
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆23Feb 15, 2025Updated last year
- ☆21Dec 22, 2020Updated 5 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 2 years ago
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆61Oct 23, 2023Updated 2 years ago
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆15Jan 3, 2023Updated 3 years ago
- Learning Task Embeddings for Teamwork Adaptation in Multi-Agent Reinforcement Learning☆14Apr 25, 2024Updated last year
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆557Feb 7, 2026Updated last week
- A collection of matrix games in JAX☆13Nov 28, 2024Updated last year
- ☆46Sep 24, 2024Updated last year
- ☆91Jan 21, 2026Updated 3 weeks ago
- Formal Contracts for Multi-Agent Reinforcement Learning☆19Oct 24, 2023Updated 2 years ago
- The Starcraft Multi-Agent challenge lite☆47Sep 13, 2024Updated last year
- SocialJax: sequential social dilemma environments☆67Nov 25, 2025Updated 2 months ago
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆28Mar 24, 2023Updated 2 years ago
- ☆20Feb 15, 2023Updated 3 years ago
- ☆19Mar 1, 2023Updated 2 years ago
- Reinforcement learning training framework for entity-gym environments.☆17Mar 18, 2024Updated last year
- Implementation of Proximal Policy Optimization in Jax+Flax☆21May 18, 2023Updated 2 years ago
- Multi-Agent Reinforcement Learning with JAX☆735Jan 14, 2026Updated last month
- A categorised list of Multi-Agent Reinforcemnt Learning (MARL) papers☆56Jan 20, 2023Updated 3 years ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆122Aug 22, 2024Updated last year
- Efficient baselines for autocurricula in JAX.☆206Aug 24, 2024Updated last year
- ☆22Nov 8, 2021Updated 4 years ago
- Fast dataset format and loader☆23Jan 2, 2026Updated last month
- Scalable Opponent Shaping Experiments in JAX☆25Apr 13, 2024Updated last year
- Codebase for the Graph-based Policy Learning algorithm, which is designed for learning policies to solve the open ad hoc teamwork problem…☆35Mar 31, 2021Updated 4 years ago
- Code for magnetic mirror descent.☆17Oct 5, 2023Updated 2 years ago
- Code for NeurIPS2023 accepted paper: Counterfactual Conservative Q Learning for Offline Multi-agent Reinforcement Learning.☆40Feb 18, 2025Updated 11 months ago
- (ICLR 2021) Learning to Represent Action Values as a Hypergraph on the Action Vertices☆23Jun 22, 2021Updated 4 years ago
- A suite of test scenarios for multi-agent reinforcement learning.☆786Feb 1, 2026Updated 2 weeks ago
- 🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX☆877Dec 15, 2025Updated 2 months ago