vivekmyers / contrastive_planning
Code for the paper "Inference via Interpolation: Contrastive Representations Provably Enable Planning and Inference"
☆35Updated 2 months ago
Related projects: ⓘ
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆51Updated 5 months ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆82Updated 9 months ago
- ☆25Updated this week
- Predictable MDP Abstraction for Unsupervised Model-Based RL (ICML 2023)☆29Updated last year
- Foundation Policies with Hilbert Representations (ICML 2024)☆65Updated 5 months ago
- Open source code for paper "Optimal Goal-Reaching Reinforcement Learning via Quasimetric Learning" ICML 2023☆42Updated 4 months ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆26Updated last month
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆27Updated last week
- Code for the ICLR 2024 spotlight paper: "Learning to Act without Actions" (introducing Latent Action Policies)☆68Updated last month
- Public code for "Reinforcement Learning from Passive Data via Latent Intentions"☆81Updated 10 months ago
- ☆23Updated 10 months ago
- a simple and scalable agent for training adaptive policies with sequence-based RL☆79Updated this week
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆49Updated 11 months ago
- ☆34Updated last year
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆71Updated 9 months ago
- Skeleton for scalable and flexible Jax RL implementations☆58Updated last year
- Learning to Modulate pre-trained Models in RL (Decision Transformer, LoRA, Fine-tuning)☆49Updated 8 months ago
- Reproduction of Dreamerv1 and v2 in pytorch for deepmind control suite☆25Updated last year
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆46Updated 3 months ago
- PyTorch Package For Quasimetric Learning☆38Updated last year
- Reinforcement Learning via Supervised Learning☆67Updated 2 years ago
- A lightweight research framework☆20Updated 6 months ago
- PWM: Policy Learning with Large World Models☆32Updated last month
- Synthetic Experience Replay☆62Updated 3 months ago
- Code release for Efficient Planning in a Compact Latent Action Space (ICLR2023) https://arxiv.org/abs/2208.10291.☆90Updated last year
- Extreme Q-Learning: Max Entropy RL without Entropy☆78Updated last year
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆83Updated 7 months ago
- [ICML 2023] Pre-train world model-based agents with different unsupervised strategies, fine-tune the agent's components selectively, and …☆31Updated 6 months ago
- Official code for "Reward-Free Curricula for Training Robust World Models", ICLR 2024.☆25Updated 7 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆59Updated last month