JAX library for MARL research
☆87Oct 28, 2023Updated 3 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"☆15Mar 1, 2023Updated 3 years ago
- ☆11May 25, 2026Updated 5 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆80Aug 18, 2024Updated 2 years ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆25Feb 15, 2025Updated last year
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)☆19May 10, 2024Updated 2 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Modular Single-file Reinfocement Learning Algorithms Library☆38May 16, 2023Updated 3 years ago
- ☆26Apr 16, 2024Updated 2 years ago
- Accelerated replay buffers in JAX☆47Sep 17, 2022Updated 4 years ago
- Scalable Opponent Shaping Experiments in JAX☆27Apr 13, 2024Updated 2 years ago
- ☆21Feb 15, 2023Updated 3 years ago
- Vectorization techniques for fast population-based training.☆58Apr 26, 2026Updated 6 months ago
- ☆102Jan 21, 2026Updated 9 months ago
- [NeurIPS 2022] Open source code for reusing prior computational work in RL.☆100Jul 5, 2023Updated 3 years ago
- Formal Contracts for Multi-Agent Reinforcement Learning☆20Oct 24, 2023Updated 3 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- ☆21Dec 22, 2020Updated 5 years ago
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆62Aug 22, 2026Updated 2 months ago
- Learning Task Embeddings for Teamwork Adaptation in Multi-Agent Reinforcement Learning☆15Apr 25, 2024Updated 2 years ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆127Aug 22, 2024Updated 2 years ago
- SocialJax: sequential social dilemma environments☆97Oct 17, 2026Updated 2 weeks ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 3 years ago
- ☆10Feb 20, 2024Updated 2 years ago
- Multi-agent simulator in Jax for research and teaching in AI & ALife☆33Apr 11, 2026Updated 6 months ago
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆14Jan 3, 2023Updated 3 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- A suite of test scenarios for multi-agent reinforcement learning.☆903Oct 13, 2026Updated 3 weeks ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆283Jun 10, 2026Updated 4 months ago
- Source code for the paper "Policy Architectures for Compositional Generalization in Control"☆30May 19, 2022Updated 4 years ago
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code …☆28Mar 24, 2023Updated 3 years ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆614Oct 27, 2026Updated last week
- Baselines for gymnax 🤖☆82Apr 3, 2023Updated 3 years ago
- Code for the ICML 2023 paper "Learning to acquire novel cognitive tasks with evolution, plasticity and meta-meta-learning"☆11Jun 3, 2023Updated 3 years ago
- Efficient baselines for autocurricula in JAX.☆217Aug 24, 2024Updated 2 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 6 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- ☆19Mar 1, 2023Updated 3 years ago
- 🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX☆939Sep 9, 2026Updated last month
- Skeleton for scalable and flexible Jax RL implementations☆102Jul 1, 2023Updated 3 years ago
- On the Feasibility of Cross-Task Transfer with Model-Based Reinforcement Learning☆16Apr 30, 2023Updated 3 years ago
- ♟️ Vectorized RL game environments in JAX☆657Mar 6, 2025Updated last year
- Repo for reproduction of sequential social dilemmas☆418Mar 6, 2025Updated last year
- Accompanying code for "Visual Encoders for Data-Efficient Imitation Learning in Modern Video Games" publication☆17Oct 15, 2026Updated 2 weeks ago