đĒ The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX
â61Aug 22, 2026Updated last month
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 5 months ago
- A collection of matrix games in JAXâ16Apr 13, 2026Updated 5 months ago
- Accelerated replay buffers in JAXâ47Sep 17, 2022Updated 4 years ago
- ⥠Flashbax: Accelerated Replay Buffers in JAXâ282Aug 27, 2026Updated last month
- Simple single file implementations of Reinforcement Learning algorithms in Juliaâ24Feb 15, 2025Updated last year
- Bare Metal GPUs on DigitalOcean Gradient AI âĸ AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- A categorised list of Multi-Agent Reinforcemnt Learning (MARL) papersâ58Jan 20, 2023Updated 3 years ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLâ125Aug 22, 2024Updated 2 years ago
- Reinforcement learning training framework for entity-gym environments.â17Mar 18, 2024Updated 2 years ago
- A tool for aggregating and plotting MARL experiment data.â88Apr 13, 2026Updated 5 months ago
- A neural network library written in jaxâ14Feb 3, 2025Updated last year
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ868Sep 9, 2026Updated last month
- Population-Based Reinforcement Learning for Combinatorial Optimizationâ88Feb 12, 2024Updated 2 years ago
- Baselines for gymnax đ¤â80Apr 3, 2023Updated 3 years ago
- Parallel hyperparameter tuning with JAXâ40Sep 30, 2026Updated last week
- AI Agents on DigitalOcean Gradient AI Platform âĸ AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- đī¸A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX âĸ End-to-End JAX RLâ423Mar 18, 2026Updated 6 months ago
- Modular Single-file Reinfocement Learning Algorithms Libraryâ38May 16, 2023Updated 3 years ago
- đ§Ŧ ManyFold: An efficient and flexible library for training and validating protein folding modelsâ80Dec 14, 2022Updated 3 years ago
- Vectorization techniques for fast population-based training.â58Apr 26, 2026Updated 5 months ago
- Datasets with baselines for Offline MARL.â232Nov 2, 2025Updated 11 months ago
- đââŦ Contextual bandits library for continuous action trees with smoothing in JAXâ72Jul 28, 2026Updated 2 months ago
- Opinionated library for managing hyperparameters and mutable state of machine learning training systems.â19Aug 4, 2023Updated 3 years ago
- COMPASS: Combinatorial Optimization with Policy Adaptation using Latent Space Searchâ47Jun 21, 2024Updated 2 years ago
- Corax: Core RL in JAXâ41Feb 22, 2024Updated 2 years ago
- Managed hosting for WordPress and PHP on Cloudways âĸ AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- A LLM-friendly framework for translating dynamical equations to gymnasium-compatible RL environments.â33Mar 18, 2026Updated 6 months ago
- đĻ A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAXâ938Sep 9, 2026Updated last month
- A toolkit for practical Human-AI cooperation researchâ15Apr 19, 2024Updated 2 years ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.â460Jun 20, 2026Updated 3 months ago
- flexible meta-learning in jaxâ16Oct 19, 2023Updated 2 years ago
- âī¸ Vectorized RL game environments in JAXâ652Mar 6, 2025Updated last year
- Library for efficient training and application of Machine Learning Interatomic Potentials (MLIP)â138Sep 25, 2026Updated 2 weeks ago
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid đī¸â344Dec 16, 2025Updated 9 months ago
- A library for deploying App on deepchain.bioâ31Sep 24, 2021Updated 5 years ago
- Proton VPN Special Offer - Get 70% off âĸ AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Minimal Decision Transformer Implementation written in Jax (Flax).â18Aug 8, 2022Updated 4 years ago
- Accelerated minigrid environments with JAXâ182Sep 10, 2026Updated last month
- â21Dec 22, 2020Updated 5 years ago
- â102Jan 21, 2026Updated 8 months ago
- Standard interface for entity based reinforcement learning environments.â40Feb 28, 2024Updated 2 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)â12Jun 15, 2023Updated 3 years ago
- AbBFN2: A flexible antibody foundation model based on Bayesian Flow Networksâ39Jun 4, 2025Updated last year