Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024
☆78Jun 4, 2024Updated last year
Alternatives and similar repositories for Recall2Imagine
Users that are interested in Recall2Imagine are comparing it to the libraries listed below
Sorting:
- Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper☆21Jul 14, 2024Updated last year
- Official Code Repo for the paper "Learning to Play Atari in a World of Tokens" accepted at ICML, 2024☆11Jun 6, 2024Updated last year
- Evaluating long-term memory of reinforcement learning algorithms☆164Jun 23, 2023Updated 2 years ago
- Clockwork VAEs in JAX/Flax☆32Jul 16, 2021Updated 4 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆20Oct 19, 2022Updated 3 years ago
- Official repository for our paper on "Action Inference by Maximising Evidence: Zero-Shot Imitation from Observation with World Models"☆13Dec 4, 2023Updated 2 years ago
- ☆36Dec 26, 2022Updated 3 years ago
- ☆19Nov 25, 2022Updated 3 years ago
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆20Jun 16, 2024Updated last year
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆87Jan 24, 2024Updated 2 years ago
- Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions☆73Aug 31, 2024Updated last year
- Revisiting Peng's Q(lambda) for Modern Reinforcement Learning☆15Jul 23, 2021Updated 4 years ago
- Official code for Slot-Transformer for Videos (STEVE)☆50Jan 9, 2023Updated 3 years ago
- Official codebase for "Privileged Sensing Scaffolds Reinforcement Learning", contains the Scaffolder algorithm and Sensory Scaffolding Su…☆33Sep 30, 2025Updated 5 months ago
- Using advances in generative modeling to learn reward functions from unlabeled videos.☆141Feb 12, 2024Updated 2 years ago
- Official Code for Neural Systematic Binder☆34Mar 27, 2023Updated 2 years ago
- off-policy RL on long sequences☆159Feb 17, 2026Updated 2 weeks ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆112Dec 5, 2023Updated 2 years ago
- Deep Hierarchical Planning from Pixels☆115Dec 21, 2022Updated 3 years ago
- Code for the ICLR 2024 spotlight paper: "Learning to Act without Actions" (introducing Latent Action Policies)☆134Jul 31, 2024Updated last year
- ☆28Mar 20, 2024Updated last year
- A list of academic papers that utilize the ManiSkill framework, categorized by year of publication.☆23May 30, 2025Updated 9 months ago
- Official code for "World Models via Policy-Guided Trajectory Diffusion", TMLR 2024☆74Mar 22, 2024Updated last year
- Fast reinforcement learning research☆61Dec 7, 2024Updated last year
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆374Feb 10, 2026Updated 3 weeks ago
- The code for the paper "A Bayesian Approach to Online Planning" published in ICML 2024.☆13Jun 17, 2024Updated last year
- Partially Observable Process Gym☆212Jun 12, 2025Updated 8 months ago
- Vintix: Action Model via In-Context Reinforcement Learning - - — ICML 2025☆45May 23, 2025Updated 9 months ago
- Meta-RL Model-Based Algorithm☆43Apr 30, 2025Updated 10 months ago
- Implementation of Dreamer v3 in pytorch.☆797Feb 21, 2026Updated last week
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Oct 12, 2023Updated 2 years ago
- ReLAx - Reinforcement Learning Applications Library☆15Feb 19, 2023Updated 3 years ago
- Code for the paper: Sparsely Changing Latent States for Prediction and Planning in Partially Observable Domains☆10Nov 12, 2021Updated 4 years ago
- Large Language Models Powered Context-aware Motion Prediction☆14Jan 12, 2026Updated last month
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Apr 19, 2024Updated last year
- PyTorch implementation of DreamerV2 model-based RL algorithm☆237Apr 26, 2023Updated 2 years ago
- Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.☆247Nov 23, 2025Updated 3 months ago
- Object-Centric-Representation Library (OCRL): This repo is to explore OCR on various downstream tasks from supervised learning tasks to R…☆12Feb 23, 2024Updated 2 years ago
- Official repository for "Investigating Pre-Training Objectives for Generalization in Visual Reinforcement Learning" (ICML 2024)☆11Sep 16, 2025Updated 5 months ago