instadeepai / sebulbaLinks
πͺ The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX
β58Updated last year
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β43Updated 2 years ago
- β81Updated 9 months ago
- β‘ Flashbax: Accelerated Replay Buffers in JAXβ246Updated 3 weeks ago
- A collection of matrix games in JAXβ11Updated 9 months ago
- β82Updated 5 months ago
- General Modules for JAXβ67Updated 4 months ago
- Baselines for gymnax π€β71Updated 2 years ago
- Accelerated minigrid environments with JAXβ143Updated 2 weeks ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorchβ52Updated 2 years ago
- Simple single-file baselines for Q-Learning in pure-GPU settingβ181Updated 5 months ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLβ114Updated last year
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselinβ¦β56Updated last week
- Vectorization techniques for fast population-based training.β56Updated 3 years ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!β232Updated 3 months ago
- β19Updated 3 months ago
- Unified Implementations of Offline Reinforcement Learning Algorithmsβ92Updated 4 months ago
- Code and data for the paper "Bridging RL Theory and Practice with the Effective Horizon"β48Updated last year
- An implementation of MuZero in JAX.β56Updated 2 years ago
- JAX implementations of core Deep RL algorithmsβ82Updated 3 years ago
- Code of the paper: Debiasing Meta-Gradient Reinforcement Learning by Learning the Outer Value Functionβ13Updated 2 years ago
- β46Updated 2 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weightsβ¦β58Updated 3 years ago
- A collection of RL algorithms written in JAX.β103Updated 3 years ago
- Contains JAX implementation of algorithms for inverse reinforcement learningβ73Updated last year
- Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"β27Updated 4 months ago
- JAX implementation of RL algorithms and vectorized environmentsβ47Updated last year
- Simple JAX Graphics Library.β36Updated 9 months ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"β100Updated 3 years ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]β110Updated last year
- An Open-Ended Agentic Simulatorβ52Updated last year