young-geng / JaxCQLView external linksLinks
Conservative Q learning in Jax
☆57Feb 7, 2023Updated 3 years ago
Alternatives and similar repositories for JaxCQL
Users that are interested in JaxCQL are comparing it to the libraries listed below
Sorting:
- Conservative Q Learning on top of SAC☆136Oct 15, 2022Updated 3 years ago
- official implementation for our paper Cal-QL: Calibrated Offline RL Pre-Training for Efficient Online Fine-Tuning (NeurIPS 2023)☆119Jul 31, 2024Updated last year
- [ICLR 2023 Oral] The official implementation of SQL and EQL in "Offline RL with No OOD Actions: In-Sample Learning via Implicit Value Reg…☆47Jul 27, 2023Updated 2 years ago
- ☆316Jan 23, 2022Updated 4 years ago
- ☆52Jan 20, 2023Updated 3 years ago
- TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.☆21Sep 24, 2025Updated 4 months ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆93Dec 1, 2024Updated last year
- GPT implementation in Flax☆18Jan 8, 2022Updated 4 years ago
- ☆10Mar 11, 2024Updated last year
- ☆58Feb 8, 2025Updated last year
- NeurIPS 2024☆14Oct 29, 2025Updated 3 months ago
- Corax: Core RL in JAX☆38Feb 22, 2024Updated last year
- Code accompanying the paper Adversarially Trained Actor Critic for Offline Reinforcement Learning by Ching-An Cheng*, Tengyang Xie*, Nan …☆72Feb 2, 2023Updated 3 years ago
- Code for MOBILE: Model-Bellman Inconsistency Penalized Offline Policy Optimization☆22Apr 17, 2024Updated last year
- ☆23Aug 19, 2022Updated 3 years ago
- Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations☆113May 27, 2024Updated last year
- Repo for Implicit Diffusion Q-Learning☆123Dec 5, 2023Updated 2 years ago
- ☆80Dec 9, 2022Updated 3 years ago
- Official PyTorch implementation of "Uncertainty-Based Offline Reinforcement Learning with Diversified Q-Ensemble" (NeurIPS'21)☆79Aug 14, 2022Updated 3 years ago
- This repository contains the implementation of the PTR algorithm described in the paper: Pre-Training for Robots: Leveraging Diverse Mult…☆30Oct 26, 2022Updated 3 years ago
- ☆15Jan 18, 2026Updated 3 weeks ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆747Oct 26, 2022Updated 3 years ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Aug 9, 2024Updated last year
- Synthetic Experience Replay☆109May 27, 2024Updated last year
- Author's PyTorch implementation of TD3+BC, a simple variant of TD3 for offline RL☆391Dec 18, 2021Updated 4 years ago
- Reinforcement Learning via Supervised Learning☆72May 16, 2022Updated 3 years ago
- ☆18May 30, 2023Updated 2 years ago
- An elegant PyTorch offline reinforcement learning library for researchers.☆383Jul 11, 2025Updated 7 months ago
- Author's PyTorch implementation of ICML'23 paper "Policy Regularization with Dataset Constraint for Offline Reinforcement Learning" for D…☆18Nov 8, 2024Updated last year
- Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.☆234Nov 23, 2025Updated 2 months ago
- A PyTorch implementation of Implicit Q-Learning☆95Oct 23, 2021Updated 4 years ago
- CREATE Environment for long-horizon physics-puzzle tasks with diverse tools☆18Nov 22, 2022Updated 3 years ago
- A benchmark for offline goal-conditioned RL and offline RL☆331Jan 14, 2026Updated 3 weeks ago
- ☆60Feb 3, 2023Updated 3 years ago
- ☆19Jun 25, 2023Updated 2 years ago
- This is code to accompany the paper "Accelerating Exploration with Unlabeled Prior Data".☆25Dec 5, 2023Updated 2 years ago
- Partially Observable Multi-Agent RL with Transformers☆17Updated this week
- Public code for "Reinforcement Learning from Passive Data via Latent Intentions"☆89Nov 19, 2023Updated 2 years ago
- The official implementation of "When Data Geometry Meets Deep Function: Generalizing Offline Reinforcement Learning" (ICLR2023)☆44Mar 6, 2023Updated 2 years ago