ikostrikov / jaxrl2Links
☆46Updated 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☆86Updated 2 years ago
- Conservative Q learning in Jax☆55Updated 2 years ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆80Updated last year
- Standalone library of frequently-used wrappers for dm_env environments.☆17Updated last year
- Fast reinforcement learning research☆61Updated 9 months ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆115Updated 3 years ago
- Source files to replicate experiments in my ICLR 2022 paper.☆70Updated last month
- On the model-based stochastic value gradient for continuous reinforcement learning☆56Updated 2 years ago
- Unified Implementations of Offline Reinforcement Learning Algorithms☆94Updated 4 months ago
- Simple maze environments using mujoco-py☆57Updated last year
- ☆43Updated 2 years ago
- [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.☆24Updated 2 years ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆100Updated 3 years ago
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Updated last year
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"☆65Updated 3 months ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆80Updated 2 years ago
- EARL: Environment for Autonomous Reinforcement Learning☆37Updated 2 years ago
- Deep Hierarchical Planning from Pixels☆107Updated 2 years ago
- Corax: Core RL in JAX☆38Updated last year
- a modular reinforcement learning library with JAX agents☆27Updated 6 months ago
- ☆17Updated last year
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆88Updated 9 months ago
- Code for "Hierarchical Skills for Efficient Exploration" HSD-3 Algorithm and Baselines☆50Updated 3 years ago
- Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations☆108Updated last year
- This repository contains the implementation of the PTR algorithm described in the paper: Pre-Training for Robots: Leveraging Diverse Mult…☆28Updated 2 years ago
- Foundation Policies with Hilbert Representations (ICML 2024)☆94Updated last year
- Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.☆183Updated 4 months ago
- Bipedal Skills Benchmark for Reinforcement Learning☆25Updated 2 years ago
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆73Updated last year