Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"
☆90Jun 4, 2024Updated last year
Alternatives and similar repositories for CrossQ
Users that are interested in CrossQ are comparing it to the libraries listed below
Sorting:
- ☆26Apr 26, 2024Updated last year
- Clean single-file implementation of offline RL algorithms in JAX☆169Nov 24, 2025Updated 3 months ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆63Jan 2, 2026Updated 2 months ago
- ☆120Feb 25, 2025Updated last year
- Source files to replicate experiments in my ICLR 2022 paper.☆71Jul 17, 2025Updated 7 months ago
- ☆10Jun 27, 2024Updated last year
- Simple single-file baselines for Q-Learning in pure-GPU setting☆235Nov 24, 2025Updated 3 months ago
- Foundation Policies with Hilbert Representations (ICML 2024)☆105Sep 29, 2025Updated 5 months ago
- Pointax: PointMaze Environment for JAX☆26Oct 22, 2025Updated 4 months ago
- A dataloader, but for JAX☆20May 17, 2024Updated last year
- Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"☆25May 5, 2024Updated last year
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆23Oct 28, 2024Updated last year
- Official code for ICML 2024 paper Reinformer: Max-Return Sequence Modeling for offline RL☆46Oct 16, 2024Updated last year
- Efficient seed-parallel implementation of "Breaking the Replay Ratio Barrier"☆27May 22, 2023Updated 2 years ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆261Oct 31, 2025Updated 4 months ago
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆20Jun 16, 2024Updated last year
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆25Jan 14, 2025Updated last year
- ☆23Aug 19, 2022Updated 3 years ago
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"☆91Nov 4, 2025Updated 3 months ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆57May 21, 2023Updated 2 years ago
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Apr 19, 2024Updated last year
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆562Updated this week
- ☆92Jan 21, 2026Updated last month
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Aug 8, 2022Updated 3 years ago
- Flax Implementation of DreamerV3 on Crafter☆18Nov 29, 2025Updated 3 months ago
- Benchmarking RL generalization in an interpretable way.☆175Nov 20, 2025Updated 3 months ago
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆86Oct 15, 2023Updated 2 years ago
- ReDo: The Dormant Neuron Phenomenon in Deep Reinforcement Learning (pytorch)☆34Oct 22, 2024Updated last year
- ☆48Nov 20, 2025Updated 3 months ago
- The official code release for "Langevin Soft Actor-Critic: Efficient Exploration through Uncertainty-Driven Critic Learning", ICLR 2025☆13May 28, 2025Updated 9 months ago
- Challenging Memory-based Deep Reinforcement Learning Agents☆111Oct 27, 2024Updated last year
- RL Environments in JAX 🌍☆864May 30, 2025Updated 9 months ago
- ☆22Jan 15, 2026Updated last month
- ☆19Mar 1, 2023Updated 3 years ago
- ☆44Jul 12, 2024Updated last year
- Drop-in environment replacements that make your RL algorithm train faster.☆21Jun 19, 2024Updated last year
- (ICLR 2024) Reverse Forward Curriculum Learning☆53Nov 28, 2024Updated last year
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆105May 17, 2022Updated 3 years ago
- Official code for TLDR: Unsupervised Goal-Conditioned RL via Temporal Distance-Aware Representations☆36Jan 24, 2026Updated last month