young-geng / JaxCQL
Conservative Q learning in Jax
☆52Updated last year
Alternatives and similar repositories for JaxCQL:
Users that are interested in JaxCQL are comparing it to the libraries listed below
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Updated 8 months ago
- [ICLR 2023 Oral] The official implementation of SQL and EQL in "Offline RL with No OOD Actions: In-Sample Learning via Implicit Value Reg…☆45Updated last year
- Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations☆98Updated 7 months ago
- ☆41Updated last year
- Skeleton for scalable and flexible Jax RL implementations☆67Updated last year
- ☆45Updated last year
- Reinforcement Learning via Supervised Learning☆69Updated 2 years ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆106Updated 2 years ago
- [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.☆24Updated last year
- Pessimistic Bootstrapping for Uncertainty-Driven Offline Reinforcement Learning☆28Updated 2 years ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆80Updated last year
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023☆19Updated last year
- Code and data for the paper "Bridging RL Theory and Practice with the Effective Horizon"☆42Updated 6 months ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆76Updated last month
- official implementation for our paper Cal-QL: Calibrated Offline RL Pre-Training for Efficient Online Fine-Tuning☆81Updated 5 months ago
- Codebase for "Uni[MASK]: Unified Inference in Sequential Decision Problems"☆54Updated 6 months ago
- Extreme Q-Learning: Max Entropy RL without Entropy☆82Updated last year
- Efficient seed-parallel implementation of "Breaking the Replay Ratio Barrier"☆22Updated last year
- Synthetic Experience Replay☆82Updated 7 months ago
- Code for MOBILE: Model-Bellman Inconsistency Penalized Offline Policy Optimization☆17Updated 9 months ago
- ☆33Updated last month
- Benchmarked implementations of Offline RL Algorithms.☆67Updated last week
- [ICML 2022] The official implementation of DWBC in "Discriminator-Weighted Offline Imitation Learning from Suboptimal Demonstrations"☆34Updated 2 years ago
- ☆17Updated 2 years ago
- ☆14Updated last year
- Simple maze environments using mujoco-py☆54Updated last year
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆63Updated 7 months ago
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆59Updated last year
- ☆16Updated 2 years ago
- Learning Laplacian Representations in Reinforcement Learning☆17Updated 4 years ago