young-geng / JaxCQLLinks
Conservative Q learning in Jax
☆56Updated 2 years ago
Alternatives and similar repositories for JaxCQL
Users that are interested in JaxCQL are comparing it to the libraries listed below
Sorting:
- ExORL: Exploratory Data for Offline Reinforcement Learning☆120Updated 3 years ago
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Updated last year
- Extreme Q-Learning: Max Entropy RL without Entropy☆87Updated 2 years ago
- ☆52Updated 2 years ago
- Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations☆112Updated last year
- Benchmarked implementations of Offline RL Algorithms.☆76Updated 9 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…☆47Updated 2 years ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆103Updated 3 years ago
- [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.☆25Updated 2 years ago
- official implementation for our paper Cal-QL: Calibrated Offline RL Pre-Training for Efficient Online Fine-Tuning (NeurIPS 2023)☆115Updated last year
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆82Updated 2 years ago
- Simple maze environments using mujoco-py☆57Updated last year
- ☆52Updated 2 years ago
- Codebase for "Uni[MASK]: Unified Inference in Sequential Decision Problems"☆57Updated last year
- ☆19Updated 2 years ago
- Reinforcement Learning via Supervised Learning☆72Updated 3 years ago
- Code accompanying the paper Adversarially Trained Actor Critic for Offline Reinforcement Learning by Ching-An Cheng*, Tengyang Xie*, Nan …☆72Updated 2 years ago
- Implemention of the Decision-Pretrained Transformer (DPT) from the paper Supervised Pretraining Can Learn In-Context Reinforcement Learni…☆80Updated last year
- Deep Hierarchical Planning from Pixels☆112Updated 3 years ago
- Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)☆69Updated 3 years ago
- Efficient seed-parallel implementation of "Breaking the Replay Ratio Barrier"☆27Updated 2 years ago
- ☆18Updated 3 years ago
- ☆31Updated 2 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆160Updated 2 years ago
- Skeleton for scalable and flexible Jax RL implementations☆92Updated 2 years ago
- Code and data for the paper "Bridging RL Theory and Practice with the Effective Horizon"☆50Updated last year
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆81Updated 2 years ago
- Repo for Implicit Diffusion Q-Learning☆120Updated 2 years ago
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)☆19Updated last year
- Code for MOBILE: Model-Bellman Inconsistency Penalized Offline Policy Optimization☆22Updated last year