đĒ The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX
â61Oct 23, 2023Updated 2 years ago
Alternatives and similar repositories for sebulba
Users that are interested in sebulba are comparing it to the libraries listed below
Sorting:
- Accelerated replay buffers in JAXâ46Sep 17, 2022Updated 3 years ago
- A collection of matrix games in JAXâ13Nov 28, 2024Updated last year
- Code of the paper: Debiasing Meta-Gradient Reinforcement Learning by Learning the Outer Value Functionâ13Nov 22, 2022Updated 3 years ago
- ⥠Flashbax: Accelerated Replay Buffers in JAXâ273Sep 22, 2025Updated 5 months ago
- Reinforcement learning training framework for entity-gym environments.â17Mar 18, 2024Updated last year
- A categorised list of Multi-Agent Reinforcemnt Learning (MARL) papersâ56Jan 20, 2023Updated 3 years ago
- Simple single file implementations of Reinforcement Learning algorithms in Juliaâ23Feb 15, 2025Updated last year
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLâ122Aug 22, 2024Updated last year
- Modular Single-file Reinfocement Learning Algorithms Libraryâ38May 16, 2023Updated 2 years ago
- Baselines for gymnax đ¤â74Apr 3, 2023Updated 2 years ago
- A tool for aggregating and plotting MARL experiment data.â83Jan 26, 2026Updated last month
- Vectorization techniques for fast population-based training.â57Aug 12, 2022Updated 3 years ago
- Opinionated library for managing hyperparameters and mutable state of machine learning training systems.â19Aug 4, 2023Updated 2 years ago
- Corax: Core RL in JAXâ39Feb 22, 2024Updated 2 years ago
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ810Dec 1, 2025Updated 3 months ago
- đī¸A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX âĸ End-to-End JAX RLâ395Feb 19, 2026Updated last week
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)â19May 10, 2024Updated last year
- A neural network library written in jaxâ13Feb 3, 2025Updated last year
- The Starcraft Multi-Agent challenge liteâ47Sep 13, 2024Updated last year
- Population-Based Reinforcement Learning for Combinatorial Optimizationâ87Feb 12, 2024Updated 2 years ago
- đââŦ Contextual bandits library for continuous action trees with smoothing in JAXâ71Oct 7, 2022Updated 3 years ago
- â92Jan 21, 2026Updated last month
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.â374Feb 10, 2026Updated 2 weeks ago
- Standard interface for entity based reinforcement learning environments.â38Feb 28, 2024Updated 2 years ago
- â21Dec 22, 2020Updated 5 years ago
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid đī¸â325Dec 16, 2025Updated 2 months ago
- flexible meta-learning in jaxâ16Oct 19, 2023Updated 2 years ago
- đĻ A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAXâ878Updated this week
- Code for Discovered Policy Optimisation (NeurIPS 2022)â12Jun 15, 2023Updated 2 years ago
- Datasets with baselines for Offline MARL.â203Nov 2, 2025Updated 4 months ago
- âī¸ Vectorized RL game environments in JAXâ588Mar 6, 2025Updated 11 months ago
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.â15Jan 3, 2023Updated 3 years ago
- A toolkit for practical Human-AI cooperation researchâ14Apr 19, 2024Updated last year
- Official Implementation of Mementoâ21Nov 19, 2024Updated last year
- PushWorld: A benchmark for manipulation planning with tools and movable obstaclesâ90Jan 13, 2026Updated last month
- đ§Ŧ ManyFold: An efficient and flexible library for training and validating protein folding modelsâ81Dec 14, 2022Updated 3 years ago
- General Modules for JAXâ72Feb 21, 2026Updated last week
- Accelerated minigrid environments with JAXâ160Oct 20, 2025Updated 4 months ago
- Docker containers of baseline agents for the Crafter environmentâ30Dec 14, 2021Updated 4 years ago