evgenii-nikishin / rl_with_resetsView external linksLinks
JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"
☆104May 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☆83May 2, 2022Updated 3 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆56May 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.☆100Oct 22, 2023Updated 2 years ago
- ☆23Aug 19, 2022Updated 3 years ago
- Standalone library of frequently-used wrappers for dm_env environments.☆18Jul 9, 2024Updated last year
- Skeleton for scalable and flexible Jax RL implementations☆95Jul 1, 2023Updated 2 years ago
- DMControl Generalization Benchmark☆188Jan 3, 2024Updated 2 years ago
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆82May 13, 2024Updated last year
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆83Jul 27, 2022Updated 3 years ago
- Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions☆73Aug 31, 2024Updated last year
- ☆278Aug 20, 2022Updated 3 years ago
- Vectorization techniques for fast population-based training.☆57Aug 12, 2022Updated 3 years ago
- ☆381Feb 13, 2023Updated 3 years ago
- Docker containers of baseline agents for the Crafter environment☆30Dec 14, 2021Updated 4 years ago
- ☆251Nov 19, 2024Updated last year
- Building blocks for productive research☆68Jan 12, 2026Updated last month
- RL Environments in JAX 🌍☆857May 30, 2025Updated 8 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
- ☆119Feb 25, 2025Updated 11 months ago
- An Empirical Investigation of Representation Learning for Imitation (EIRLI), NeurIPS'21☆36Mar 4, 2023Updated 2 years ago
- 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…☆35Oct 24, 2025Updated 3 months ago
- Code for MOBILE: Model-Bellman Inconsistency Penalized Offline Policy Optimization☆22Apr 17, 2024Updated last year
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆866Aug 12, 2024Updated last year
- Corax: Core RL in JAX☆38Feb 22, 2024Updated last year
- ExORL: Exploratory Data for Offline Reinforcement Learning☆128Feb 8, 2022Updated 4 years ago