đĒ The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX
â62Aug 22, 2026Updated 2 months 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:
- Code of the paper: Debiasing Meta-Gradient Reinforcement Learning by Learning the Outer Value Functionâ13Apr 13, 2026Updated 6 months ago
- Accelerated replay buffers in JAXâ47Sep 17, 2022Updated 4 years ago
- ⥠Flashbax: Accelerated Replay Buffers in JAXâ282Aug 27, 2026Updated 2 months ago
- A collection of matrix games in JAXâ16Apr 13, 2026Updated 6 months ago
- Simple single file implementations of Reinforcement Learning algorithms in Juliaâ25Feb 15, 2025Updated last year
- Serverless GPU API endpoints on Runpod - Get Bonus Credits âĸ AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLâ127Aug 22, 2024Updated 2 years ago
- A categorised list of Multi-Agent Reinforcemnt Learning (MARL) papersâ58Jan 20, 2023Updated 3 years ago
- Vectorization techniques for fast population-based training.â58Apr 26, 2026Updated 6 months ago
- A tool for aggregating and plotting MARL experiment data.â89Apr 13, 2026Updated 6 months ago
- A JAX-Native Interface for Reinforcement Learning Environmentsâ10Oct 6, 2025Updated last year
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ875Oct 16, 2026Updated 2 weeks ago
- Reinforcement learning training framework for entity-gym environments.â17Mar 18, 2024Updated 2 years ago
- đī¸A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX âĸ End-to-End JAX RLâ425Mar 18, 2026Updated 7 months ago
- Population-Based Reinforcement Learning for Combinatorial Optimizationâ88Feb 12, 2024Updated 2 years ago
- Open source password manager - Proton Pass âĸ AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- Baselines for gymnax đ¤â82Apr 3, 2023Updated 3 years ago
- đ§Ŧ ManyFold: An efficient and flexible library for training and validating protein folding modelsâ80Dec 14, 2022Updated 3 years ago
- Accelerated Quality-Diversityâ364Oct 30, 2025Updated last year
- A neural network library written in jaxâ14Feb 3, 2025Updated last year
- PPO (Proximal Policy Optimization) agent for training on Atari environments using C++ with libtorch, integrated with the Arcade Learning âĻâ10Jun 14, 2026Updated 4 months ago
- COMPASS: Combinatorial Optimization with Policy Adaptation using Latent Space Searchâ47Jun 21, 2024Updated 2 years ago
- A toolkit for practical Human-AI cooperation researchâ15Apr 19, 2024Updated 2 years ago
- Opinionated library for managing hyperparameters and mutable state of machine learning training systems.â19Aug 4, 2023Updated 3 years ago
- đââŦ Contextual bandits library for continuous action trees with smoothing in JAXâ72Jul 28, 2026Updated 3 months 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.
- Corax: Core RL in JAXâ41Feb 22, 2024Updated 2 years ago
- Datasets with baselines for Offline MARL.â232Nov 2, 2025Updated 11 months ago
- âī¸ Vectorized RL game environments in JAXâ657Mar 6, 2025Updated last year
- Parallel hyperparameter tuning with JAXâ40Sep 30, 2026Updated last month
- Modular Single-file Reinfocement Learning Algorithms Libraryâ38May 16, 2023Updated 3 years ago
- đĻ A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAXâ939Sep 9, 2026Updated last month
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.â468Updated this week
- Minimal Decision Transformer Implementation written in Jax (Flax).â18Aug 8, 2022Updated 4 years ago
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid đī¸â346Dec 16, 2025Updated 10 months ago
- Managed Database hosting by DigitalOcean âĸ AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Accelerated minigrid environments with JAXâ184Sep 10, 2026Updated last month
- Standard interface for entity based reinforcement learning environments.â40Feb 28, 2024Updated 2 years ago
- A LLM-friendly framework for translating dynamical equations to gymnasium-compatible RL environments.â33Mar 18, 2026Updated 7 months ago
- â98Feb 16, 2026Updated 8 months ago
- A Jax/Stax implementation of the general meta learning paper: Oh, J., Hessel, M., Czarnecki, W.M., Xu, Z., van Hasselt, H.P., Singh, S. aâĻâ23Dec 22, 2020Updated 5 years ago
- â102Jan 21, 2026Updated 9 months ago
- Efficient baselines for autocurricula in JAX.â217Aug 24, 2024Updated 2 years ago