JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"
☆105May 17, 2022Updated 3 years ago
Alternatives and similar repositories for rl_with_resets
Users that are interested in rl_with_resets are comparing it to the libraries listed below
Sorting:
- ☆52Jan 20, 2023Updated 3 years ago
- Efficient seed-parallel implementation of "Breaking the Replay Ratio Barrier"☆27May 22, 2023Updated 2 years ago
- Code-base for the paper Spectral Normalisation for Deep Reinforcement Learning: An Optimisation Perspective.☆11Jun 26, 2021Updated 4 years ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Aug 8, 2022Updated 3 years ago
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- Modular Single-file Reinfocement Learning Algorithms Library☆38May 16, 2023Updated 2 years ago
- Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations☆113May 27, 2024Updated last year
- ☆13Aug 9, 2022Updated 3 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆20Oct 19, 2022Updated 3 years ago
- Source files to replicate experiments in my ICLR 2022 paper.☆71Jul 17, 2025Updated 7 months ago
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 3 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆57May 21, 2023Updated 2 years ago
- Author's PyTorch implementation of TD7 for online and offline RL☆161Sep 12, 2023Updated 2 years ago
- Reinforcement learning library in JAX.☆101Oct 22, 2023Updated 2 years ago
- Standalone library of frequently-used wrappers for dm_env environments.☆18Jul 9, 2024Updated last year
- ☆23Aug 19, 2022Updated 3 years ago
- Skeleton for scalable and flexible Jax RL implementations☆96Jul 1, 2023Updated 2 years ago
- DMControl Generalization Benchmark☆188Jan 3, 2024Updated 2 years ago
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆83Jul 27, 2022Updated 3 years ago
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆82May 13, 2024Updated last year
- Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions☆73Aug 31, 2024Updated last year
- ☆279Aug 20, 2022Updated 3 years ago
- Vectorization techniques for fast population-based training.☆57Aug 12, 2022Updated 3 years ago
- ☆385Feb 13, 2023Updated 3 years ago
- Docker containers of baseline agents for the Crafter environment☆30Dec 14, 2021Updated 4 years ago
- ☆254Nov 19, 2024Updated last year
- Building blocks for productive research☆69Jan 12, 2026Updated last month
- RL Environments in JAX 🌍☆868May 30, 2025Updated 9 months ago
- DrQ-v2: Improved Data-Augmented Reinforcement Learning☆431May 31, 2022Updated 3 years ago
- Experiment. Plot. Tabulate.☆73Aug 22, 2024Updated last year
- ☆46Sep 24, 2024Updated last year
- ☆122Feb 25, 2025Updated last year
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Sep 4, 2023Updated 2 years ago
- Code for the paper "Stable Gradients for Stable Learning at Scale in Deep Reinforcement Learning". Great performance in many environments…☆36Oct 24, 2025Updated 4 months ago
- An Empirical Investigation of Representation Learning for Imitation (EIRLI), NeurIPS'21☆37Mar 4, 2023Updated 3 years ago
- Code for MOBILE: Model-Bellman Inconsistency Penalized Offline Policy Optimization☆23Apr 17, 2024Updated last year
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆867Aug 12, 2024Updated last year
- Corax: Core RL in JAX☆40Feb 22, 2024Updated 2 years ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆129Feb 8, 2022Updated 4 years ago