A framework for Reinforcement Learning research.
☆248Feb 9, 2026Updated 3 weeks ago
Alternatives and similar repositories for RL-X
Users that are interested in RL-X are comparing it to the libraries listed below
Sorting:
- ☆210May 15, 2025Updated 9 months ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆263Oct 31, 2025Updated 4 months ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆396Updated this week
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆562Feb 25, 2026Updated last week
- Simple single-file baselines for Q-Learning in pure-GPU setting☆237Nov 24, 2025Updated 3 months ago
- RL Environments in JAX 🌍☆868May 30, 2025Updated 9 months ago
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆28Mar 24, 2023Updated 2 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆57May 21, 2023Updated 2 years ago
- Really Fast End-to-End Jax RL Implementations☆1,028Sep 9, 2024Updated last year
- Modular Single-file Reinfocement Learning Algorithms Library☆38May 16, 2023Updated 2 years ago
- Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.☆247Nov 23, 2025Updated 3 months ago
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.☆233Feb 26, 2026Updated last week
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆273Sep 22, 2025Updated 5 months ago
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆15Jan 3, 2023Updated 3 years ago
- [NeurIPS 2022] Open source code for reusing prior computational work in RL.☆100Jul 5, 2023Updated 2 years ago
- ☆122Feb 25, 2025Updated last year
- Imitation learning benchmark focusing on complex locomotion tasks using MuJoCo.☆1,342Updated this week
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆374Feb 10, 2026Updated 3 weeks ago
- ☆254Nov 19, 2024Updated last year
- Simple JAX Graphics Library.☆36Nov 3, 2024Updated last year
- Contains JAX implementation of algorithms for inverse reinforcement learning☆74Aug 18, 2024Updated last year
- Clean single-file implementation of offline RL algorithms in JAX☆170Nov 24, 2025Updated 3 months ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Aug 8, 2022Updated 3 years ago
- ☆721Sep 18, 2025Updated 5 months ago
- Code to accompany the paper "The Information Geometry of Unsupervised Reinforcement Learning"☆20Oct 6, 2021Updated 4 years ago
- This repository contains the code base for the paper "Gait in Eight: Efficient On-Robot Learning for Omnidirectional Quadruped Locomotion…☆13Mar 8, 2025Updated 11 months ago
- Vectorization techniques for fast population-based training.☆57Aug 12, 2022Updated 3 years ago
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆810Dec 1, 2025Updated 3 months ago
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆677Aug 22, 2025Updated 6 months ago
- Code for "TD-MPC2: Scalable, Robust World Models for Continuous Control"☆764May 21, 2025Updated 9 months ago
- Reinforcement Learning inside a 3D soccer simulation☆37Sep 15, 2024Updated last year
- ☆39Feb 4, 2026Updated last month
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆101Jan 23, 2025Updated last year
- Toolkit of Causal Model-based Reinforcement Learning.☆33Jun 5, 2023Updated 2 years ago
- Jaxplorer is a Jax reinforcement learning (RL) framework for exploring new ideas.☆13Jul 19, 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
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- Author's implementation of ReBRAC, a minimalist improvement upon TD3+BC☆62Aug 3, 2023Updated 2 years ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆64Jan 2, 2026Updated 2 months ago