off-policy RL on long sequences
☆159Feb 17, 2026Updated last week
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
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆25Jan 14, 2025Updated last year
- code for polite☆11Feb 28, 2024Updated 2 years ago
- ☆10Jun 27, 2024Updated last year
- ☆92Jan 21, 2026Updated last month
- (ICLR 2024) Reverse Forward Curriculum Learning☆53Nov 28, 2024Updated last year
- Foundation Policies with Hilbert Representations (ICML 2024)☆105Sep 29, 2025Updated 5 months ago
- Partially Observable Process Gym☆212Jun 12, 2025Updated 8 months ago
- Official Implementation for "In-Context Reinforcement Learning from Noise Distillation"☆34Sep 18, 2024Updated last year
- ☆13May 21, 2023Updated 2 years ago
- Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"☆25May 5, 2024Updated last year
- ☆46Jan 11, 2024Updated 2 years ago
- Official code for ICML 2024 paper Reinformer: Max-Return Sequence Modeling for offline RL☆46Oct 16, 2024Updated last year
- ☆16Jan 13, 2023Updated 3 years ago
- [ICLR 2024 oral] Pre-Training Goal-based Models for Sample-Efficient Reinforcement Learning☆29Mar 1, 2024Updated 2 years ago
- Official implementation of GROOT, CoRL 2023☆67Nov 4, 2023Updated 2 years ago
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️☆325Dec 16, 2025Updated 2 months ago
- Challenging Memory-based Deep Reinforcement Learning Agents☆111Oct 27, 2024Updated last year
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆101Jan 23, 2025Updated last year
- [ICLR 2023] Choreographer: a world-model-based agent that discovers and learns unsupervised skills in latent imagination, and it's able t…☆42Jun 18, 2024Updated last year
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Apr 19, 2024Updated last year
- Awesome In-Context RL: A curated list of In-Context Reinforcement Learning - - —☆283Sep 8, 2025Updated 5 months ago
- ☆17Aug 2, 2022Updated 3 years ago
- The official implementation of "Horizon Reduction Makes RL Scalable"☆181Aug 2, 2025Updated 6 months ago
- Evaluating long-term memory of reinforcement learning algorithms☆164Jun 23, 2023Updated 2 years 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
- Baselines and Datasets for Pokémon Showdown RL☆86Feb 6, 2026Updated 3 weeks ago
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆78Jun 4, 2024Updated last year
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Feb 20, 2026Updated last week
- PWM: Policy Learning with Large World Models☆64Aug 4, 2025Updated 6 months ago
- POPGym Library in JAX☆12Apr 15, 2024Updated last year
- ☆34Mar 11, 2025Updated 11 months ago
- ☆19Mar 1, 2023Updated 2 years ago
- ☆44Jul 12, 2024Updated last year
- Official implementation for the paper "Offline Meta RL - Identifiability Challenges and Effective Data Collection Strategies", NeurIPS 20…☆31Nov 23, 2021Updated 4 years ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆88Jun 4, 2024Updated last year
- Drop-in environment replacements that make your RL algorithm train faster.☆21Jun 19, 2024Updated last year
- Vision package for robot manipulation and learning research☆25Apr 21, 2024Updated last year
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆17Nov 24, 2025Updated 3 months ago