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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Official code for "A General Learning Framework for Open Ad Hoc Teamwork Using Graph-based Policy Learning"☆17Mar 1, 2023Updated 3 years ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆75Aug 18, 2024Updated last year
- ☆20Feb 15, 2023Updated 3 years ago
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)☆19May 10, 2024Updated last year
- Modular Single-file Reinfocement Learning Algorithms Library☆38May 16, 2023Updated 2 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- Codebase for BRDiv: Diverse teammate generation for ad hoc teamwork☆13May 2, 2024Updated last year
- [NeurIPS 2022] Open source code for reusing prior computational work in RL.☆99Jul 5, 2023Updated 2 years ago
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- Formal Contracts for Multi-Agent Reinforcement Learning☆19Oct 24, 2023Updated 2 years ago
- Learning Task Embeddings for Teamwork Adaptation in Multi-Agent Reinforcement Learning☆13Apr 25, 2024Updated last year
- ☆94Jan 21, 2026Updated 2 months ago
- SocialJax: sequential social dilemma environments☆73Updated this week
- ☆21Dec 22, 2020Updated 5 years ago
- Code for NeurIPS2023 accepted paper: Counterfactual Conservative Q Learning for Offline Multi-agent Reinforcement Learning.☆41Feb 18, 2025Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- 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☆62Oct 23, 2023Updated 2 years ago
- Multi-Agent Reinforcement Learning with JAX☆778Jan 14, 2026Updated 2 months ago
- ☆10Feb 20, 2024Updated 2 years ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆576Feb 25, 2026Updated last month
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆15Jan 3, 2023Updated 3 years ago
- 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
- ☆19Mar 1, 2023Updated 3 years ago
- 🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX☆892Feb 26, 2026Updated last month
- NordVPN Special Discount Offer • AdSave on top-rated NordVPN 1 or 2-year plans with secure browsing, privacy protection, and support for for all major platforms.
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆23Feb 15, 2025Updated last year
- Scalable Opponent Shaping Experiments in JAX☆25Apr 13, 2024Updated last year
- Efficient baselines for autocurricula in JAX.☆211Aug 24, 2024Updated last year
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 5 years ago
- Repo for running various baselines with Behavior-1K☆34Nov 7, 2025Updated 4 months ago
- A collection of matrix games in JAX☆13Nov 28, 2024Updated last year
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆123Aug 22, 2024Updated last year
- The Starcraft Multi-Agent challenge lite☆47Sep 13, 2024Updated last year
- Multi-agent simulator in Jax for research and teaching in AI & ALife☆31Mar 22, 2026Updated last week
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- ☆22Nov 8, 2021Updated 4 years ago
- Baselines for gymnax 🤖☆75Apr 3, 2023Updated 2 years ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆265Oct 31, 2025Updated 4 months ago
- A suite of test scenarios for multi-agent reinforcement learning.☆806Updated this week
- A categorised list of Multi-Agent Reinforcemnt Learning (MARL) papers☆57Jan 20, 2023Updated 3 years ago
- A collection of RL algorithms written in JAX.☆105Jul 5, 2022Updated 3 years ago
- A web based platform for collecting human actions in reinforcement learning environments☆31Sep 10, 2025Updated 6 months ago