đĒ 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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- 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
- Accelerated replay buffers in JAXâ46Sep 17, 2022Updated 3 years ago
- ⥠Flashbax: Accelerated Replay Buffers in JAXâ274Sep 22, 2025Updated 6 months ago
- Simple single file implementations of Reinforcement Learning algorithms in Juliaâ23Feb 15, 2025Updated last year
- 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
- Reinforcement learning training framework for entity-gym environments.â17Mar 18, 2024Updated 2 years ago
- A tool for aggregating and plotting MARL experiment data.â84Jan 26, 2026Updated last month
- A neural network library written in jaxâ13Feb 3, 2025Updated last year
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ817Mar 9, 2026Updated 2 weeks ago
- Baselines for gymnax đ¤â75Apr 3, 2023Updated 2 years ago
- Population-Based Reinforcement Learning for Combinatorial Optimizationâ87Feb 12, 2024Updated 2 years ago
- đī¸A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX âĸ End-to-End JAX RLâ397Mar 1, 2026Updated 3 weeks ago
- đ§Ŧ ManyFold: An efficient and flexible library for training and validating protein folding modelsâ81Dec 14, 2022Updated 3 years ago
- Vectorization techniques for fast population-based training.â57Aug 12, 2022Updated 3 years ago
- Modular Single-file Reinfocement Learning Algorithms Libraryâ38May 16, 2023Updated 2 years ago
- đââŦ Contextual bandits library for continuous action trees with smoothing in JAXâ71Oct 7, 2022Updated 3 years ago
- Opinionated library for managing hyperparameters and mutable state of machine learning training systems.â19Aug 4, 2023Updated 2 years ago
- Datasets with baselines for Offline MARL.â204Nov 2, 2025Updated 4 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.â381Feb 10, 2026Updated last month
- Official Implementation of Mementoâ21Nov 19, 2024Updated last year
- đĻ A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAXâ887Feb 26, 2026Updated 3 weeks ago
- Corax: Core RL in JAXâ41Feb 22, 2024Updated 2 years ago
- A toolkit for practical Human-AI cooperation researchâ14Apr 19, 2024Updated last year
- flexible meta-learning in jaxâ16Oct 19, 2023Updated 2 years ago
- âī¸ Vectorized RL game environments in JAXâ593Mar 6, 2025Updated last year
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid đī¸â326Dec 16, 2025Updated 3 months ago
- Library for efficient training and application of Machine Learning Interatomic Potentials (MLIP)â84Jan 7, 2026Updated 2 months ago
- Parallel hyperparameter tuning with JAXâ39Updated this week
- A library for deploying App on deepchain.bioâ31Sep 24, 2021Updated 4 years ago
- Accelerated minigrid environments with JAXâ163Oct 20, 2025Updated 5 months ago
- â94Jan 21, 2026Updated 2 months ago
- Minimal Decision Transformer Implementation written in Jax (Flax).â17Aug 8, 2022Updated 3 years ago
- â21Dec 22, 2020Updated 5 years ago
- Standard interface for entity based reinforcement learning environments.â38Feb 28, 2024Updated 2 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)â12Jun 15, 2023Updated 2 years ago
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)â19May 10, 2024Updated last year
- Simple single-file baselines for Q-Learning in pure-GPU settingâ237Nov 24, 2025Updated 3 months ago