MTM Masked Trajectory Models for Prediction, Representation, and Control.
☆162Dec 16, 2025Updated 2 months ago
Alternatives and similar repositories for mtm
Users that are interested in mtm are comparing it to the libraries listed below
Sorting:
- Codebase for "Uni[MASK]: Unified Inference in Sequential Decision Problems"☆57Jul 3, 2024Updated last year
- Code for "Masked Autoencoding for Scalable and Generalizable Decision Making". NeurIPS 2022☆47Mar 12, 2024Updated last year
- Code release for the paper "Goal Representations for Instruction Following: A Semi-Supervised Language Interface to Control"☆17Apr 9, 2024Updated last year
- ☆364May 1, 2023Updated 2 years ago
- PyTorch implementation for all methods and environments in the paper "MIMEx: Intrinsic Rewards from Masked Input Modeling"☆16May 17, 2023Updated 2 years ago
- From Play to Policy: Conditional Behavior Generation from Uncurated Robot Data☆56Mar 1, 2023Updated 3 years ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆129Feb 8, 2022Updated 4 years ago
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆42Jul 16, 2023Updated 2 years ago
- Code and website for Behavior Transformers: Cloning k modes with one stone.☆140May 25, 2023Updated 2 years ago
- Online Decision Transformer☆274Jan 22, 2024Updated 2 years ago
- Implementation of BIMRL: Brain Inspired Meta Reinforcement Learning - Roozbeh Razavi et al. (IROS 2022)☆10Dec 1, 2022Updated 3 years ago
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆82May 13, 2024Updated last year
- Open source code for paper "Denoised MDPs: Learning World Models Better Than the World Itself"☆137Aug 15, 2023Updated 2 years ago
- ☆80Dec 9, 2022Updated 3 years ago
- Public code for "Reinforcement Learning from Passive Data via Latent Intentions"☆89Nov 19, 2023Updated 2 years ago
- ☆17Oct 12, 2023Updated 2 years ago
- ☆36May 24, 2023Updated 2 years ago
- Pre-training Reusable Representations for Robotic Manipulation Using Diverse Human Video Data☆366Mar 21, 2023Updated 2 years ago
- ☆129Jul 30, 2024Updated last year
- ☆360Oct 12, 2022Updated 3 years ago
- Codebase for ICLR 2023 paper, "SMART: Self-supervised Multi-task pretrAining with contRol Transformers"☆54Jan 26, 2024Updated 2 years ago
- Official codebase for Decision Transformer: Reinforcement Learning via Sequence Modeling.☆2,770Apr 29, 2024Updated last year
- Code for the paper "Planning with Diffusion for Flexible Behavior Synthesis"☆1,243Jul 18, 2024Updated last year
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆82Mar 9, 2023Updated 2 years ago
- DrQ-v2: Improved Data-Augmented Reinforcement Learning☆431May 31, 2022Updated 3 years ago
- Transformer-based World Models☆89Apr 4, 2023Updated 2 years ago
- ☆153Oct 12, 2021Updated 4 years ago
- TACO-RL: Latent Plans for Task-Agnostic Offline Reinforcement Learning☆30Jan 26, 2023Updated 3 years ago
- ☆15Sep 7, 2022Updated 3 years ago
- [ICLR 2024 oral] Pre-Training Goal-based Models for Sample-Efficient Reinforcement Learning☆29Mar 1, 2024Updated 2 years ago
- Web Based Visualizer for Simulation Environments☆392May 26, 2024Updated last year
- Official code repository for Prompt-DT.☆121Aug 3, 2022Updated 3 years ago
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆34Mar 29, 2023Updated 2 years ago
- ☆34Aug 7, 2024Updated last year
- ☆385Feb 13, 2023Updated 3 years ago
- PushWorld: A benchmark for manipulation planning with tools and movable obstacles☆90Jan 13, 2026Updated last month
- Implementation of ICML 2023 paper: Future-conditioned Unsupervised Pretraining for Decision Transformer☆29Jul 25, 2023Updated 2 years ago
- Code release for Efficient Planning in a Compact Latent Action Space (ICLR2023) https://arxiv.org/abs/2208.10291.☆112May 12, 2023Updated 2 years ago
- ☆41Oct 16, 2023Updated 2 years ago