off-policy RL on long sequences
☆161Mar 8, 2026Updated 2 weeks ago
Alternatives and similar repositories for amago
Users that are interested in amago are comparing it to the libraries listed below
Sorting:
- Implemention of the Decision-Pretrained Transformer (DPT) from the paper Supervised Pretraining Can Learn In-Context Reinforcement Learni…☆78May 28, 2024Updated last year
- ☆16Aug 2, 2022Updated 3 years ago
- ☆13May 21, 2023Updated 2 years ago
- ☆47Jan 11, 2024Updated 2 years ago
- Foundation Policies with Hilbert Representations (ICML 2024)☆105Sep 29, 2025Updated 5 months ago
- Official implementation of GROOT, CoRL 2023☆68Nov 4, 2023Updated 2 years ago
- ☆16Jan 13, 2023Updated 3 years ago
- Partially Observable Process Gym☆213Jun 12, 2025Updated 9 months ago
- Official Implementation for "In-Context Reinforcement Learning from Noise Distillation"☆34Sep 18, 2024Updated last year
- Awesome In-Context RL: A curated list of In-Context Reinforcement Learning - - —☆287Sep 8, 2025Updated 6 months ago
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆27Jan 14, 2025Updated last year
- ☆10Jun 27, 2024Updated last year
- 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
- (ICLR 2024) Reverse Forward Curriculum Learning☆53Nov 28, 2024Updated last year
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Apr 19, 2024Updated last year
- Official code for ICML 2024 paper Reinformer: Max-Return Sequence Modeling for offline RL☆47Oct 16, 2024Updated last year
- code for polite☆11Feb 28, 2024Updated 2 years ago
- Model-Based Uncertainty in Value Functions (AISTATS2023)☆16Feb 28, 2023Updated 3 years ago
- The official implementation of "Horizon Reduction Makes RL Scalable"☆182Aug 2, 2025Updated 7 months ago
- Baselines and Datasets for Pokémon Showdown RL☆92Updated this week
- ☆94Jan 21, 2026Updated 2 months ago
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️☆326Dec 16, 2025Updated 3 months ago
- Official implementation for the paper "Offline Meta RL - Identifiability Challenges and Effective Data Collection Strategies", NeurIPS 20…☆31Nov 23, 2021Updated 4 years ago
- Vision package for robot manipulation and learning research☆25Apr 21, 2024Updated last year
- Official PyTorch implementation of Doduo: Dense Visual Correspondence from Unsupervised Semantic-Aware Flow☆44Feb 1, 2024Updated 2 years ago
- Challenging Memory-based Deep Reinforcement Learning Agents☆111Oct 27, 2024Updated last year
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆103Jan 23, 2025Updated last year
- Evaluating long-term memory of reinforcement learning algorithms☆165Jun 23, 2023Updated 2 years ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Feb 20, 2026Updated last month
- PWM: Policy Learning with Large World Models☆64Aug 4, 2025Updated 7 months ago
- ☆45Jul 12, 2024Updated last year
- ☆34Mar 11, 2025Updated last year
- Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"☆25May 5, 2024Updated last year
- Official code repository for Prompt-DT.☆121Aug 3, 2022Updated 3 years ago
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆79Jun 4, 2024Updated last year
- ☆87Aug 4, 2025Updated 7 months ago
- Code for "Hierarchical World Models as Visual Whole-Body Humanoid Controllers"☆204Sep 18, 2025Updated 6 months ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆129Feb 8, 2022Updated 4 years ago
- Implements the Messenger environment and EMMA model.☆25Jun 14, 2023Updated 2 years ago