ikostrikov / jaxrl2Links
☆48Updated 2 years ago
Alternatives and similar repositories for jaxrl2
Users that are interested in jaxrl2 are comparing it to the libraries listed below
Sorting:
- ☆23Updated 3 years ago
- Skeleton for scalable and flexible Jax RL implementations☆87Updated 2 years ago
- Conservative Q learning in Jax☆55Updated 2 years ago
- Fast reinforcement learning research☆61Updated 10 months ago
- Standalone library of frequently-used wrappers for dm_env environments.☆17Updated last year
- On the model-based stochastic value gradient for continuous reinforcement learning☆57Updated 2 years ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆79Updated last year
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆81Updated 2 years ago
- Simple maze environments using mujoco-py☆56Updated last year
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆102Updated 3 years ago
- Corax: Core RL in JAX☆38Updated last year
- ExORL: Exploratory Data for Offline Reinforcement Learning☆116Updated 3 years ago
- EARL: Environment for Autonomous Reinforcement Learning☆37Updated 2 years ago
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Updated last year
- Deep Hierarchical Planning from Pixels☆109Updated 2 years ago
- Efficient seed-parallel implementation of "Breaking the Replay Ratio Barrier"☆27Updated 2 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆149Updated 2 years ago
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆81Updated 3 years ago
- ☆18Updated last year
- Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.☆190Updated 2 weeks ago
- ☆48Updated 2 years ago
- Change-Based Exploration Transfer☆35Updated 3 years ago
- [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.☆24Updated 2 years ago
- Clean single-file implementation of offline RL algorithms in JAX☆158Updated 10 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆72Updated last year
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆78Updated 2 years ago
- ☆18Updated 5 months ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆60Updated last month
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆85Updated last year
- Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations☆111Updated last year