☆44Jul 12, 2024Updated last year
Alternatives and similar repositories for rlbase_stable
Users that are interested in rlbase_stable are comparing it to the libraries listed below
Sorting:
- Object-Centric-Representation Library (OCRL): This repo is to explore OCR on various downstream tasks from supervised learning tasks to R…☆12Feb 23, 2024Updated 2 years ago
- Skeleton for scalable and flexible Jax RL implementations☆96Jul 1, 2023Updated 2 years ago
- speed-running solving robot manipulation tasks☆24Oct 31, 2024Updated last year
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆261Oct 31, 2025Updated 4 months ago
- Revisiting Peng's Q(lambda) for Modern Reinforcement Learning☆15Jul 23, 2021Updated 4 years ago
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆25Jan 14, 2025Updated last year
- Clean single-file implementation of offline RL algorithms in JAX☆169Nov 24, 2025Updated 3 months ago
- ☆31Jun 21, 2024Updated last year
- Code release for "Training Robots to Evaluate Robots" (CoRL'22, Best Paper Award)☆17Feb 15, 2023Updated 3 years ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆41Jun 6, 2024Updated last year
- Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"☆25May 5, 2024Updated last year
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆90Jun 4, 2024Updated last year
- ☆92Jan 21, 2026Updated last month
- Simple single-file baselines for Q-Learning in pure-GPU setting☆235Nov 24, 2025Updated 3 months ago
- Official release of the DMControl Generalization Benchmark 2 (DMC-GB2)☆22Jul 21, 2025Updated 7 months ago
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Sep 4, 2023Updated 2 years ago
- Experiments Notebook of "Understanding the Skill Gap in Recurrent Language Models: The Role of the Gather-and-Aggregate Mechanism"☆14Apr 30, 2025Updated 10 months ago
- Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...☆14May 28, 2025Updated 9 months ago
- ☆10Jun 27, 2024Updated last year
- Official code for Slot-Transformer for Videos (STEVE)☆50Jan 9, 2023Updated 3 years ago
- Reimplementation of facebook's DinoV2 in JAX. Inference (with pretrained weights) only; training is unsupported.☆12Jun 25, 2024Updated last year
- Subtask-Aware Visual Reward Learning from Segmented Demonstrations (ICLR 2025 accepted)☆18Apr 11, 2025Updated 10 months ago
- ☆22Sep 16, 2025Updated 5 months ago
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆17Nov 24, 2025Updated 3 months ago
- ☆52Jan 20, 2023Updated 3 years ago
- Reinforcement Learning inside a 3D soccer simulation☆37Sep 15, 2024Updated last year
- ☆16Jul 16, 2024Updated last year
- Scalable metrics logging and analysis☆18Apr 9, 2025Updated 10 months ago
- [ICML 2025 GenBio Workshop] Official Implementation for "Electrostatics from Laplacian Eigenbasis for Neural Network Interatomic Potentia…☆17Jun 12, 2025Updated 8 months ago
- ☆18Apr 28, 2024Updated last year
- Pointax: PointMaze Environment for JAX☆26Oct 22, 2025Updated 4 months ago
- [ICLR 2025] Drama: Mamba-Enabled Model-Based Reinforcement Learning Is Sample and Parameter Efficien. The frist Mamba/Mamba2 MBRL agent.☆25Feb 5, 2025Updated last year
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Feb 20, 2026Updated last week
- Deep memory and sequence models in JAX☆23Jan 15, 2026Updated last month
- E-book for AIS1003☆18Oct 27, 2023Updated 2 years ago
- Code for "Baba Is AI: Break the Rules to Beat the Benchmark"☆41Sep 3, 2025Updated 6 months ago
- Public code for "Reinforcement Learning from Passive Data via Latent Intentions"☆89Nov 19, 2023Updated 2 years ago
- ☆15Sep 4, 2025Updated 5 months ago
- V-MPO torch version with DMLab30 and GTrXL☆13Mar 1, 2021Updated 5 years ago