Codebase for "Uni[MASK]: Unified Inference in Sequential Decision Problems"
☆57Jul 3, 2024Updated last year
Alternatives and similar repositories for uniMASK
Users that are interested in uniMASK are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- MTM Masked Trajectory Models for Prediction, Representation, and Control.☆163Dec 16, 2025Updated 3 months ago
- ☆14May 31, 2022Updated 3 years ago
- From Play to Policy: Conditional Behavior Generation from Uncurated Robot Data☆56Mar 1, 2023Updated 3 years ago
- Code for Abstract-to-Executable Trajectory Translation for One Shot Task Generalization (ICML 2023)☆23May 12, 2023Updated 2 years ago
- Asymmetric methods for partially observable reinforcement learning☆10Jun 9, 2025Updated 9 months ago
- Code for "Masked Autoencoding for Scalable and Generalizable Decision Making". NeurIPS 2022☆47Mar 12, 2024Updated 2 years ago
- Code to accompany the paper "The Information Geometry of Unsupervised Reinforcement Learning"☆20Oct 6, 2021Updated 4 years ago
- ☆10Oct 26, 2022Updated 3 years ago
- (NeurIPS '22) LISA: Learning Interpretable Skill Abstractions - A framework for unsupervised skill learning using Imitation☆29Feb 22, 2023Updated 3 years ago
- This is the code for Q-value Path Decomposition for Deep Multiagent Reinforcement Learning (NeurIPS 2019).☆12May 20, 2019Updated 6 years ago
- Implementation of ICML 2023 paper: Future-conditioned Unsupervised Pretraining for Decision Transformer☆29Jul 25, 2023Updated 2 years ago
- Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)☆70Aug 8, 2022Updated 3 years ago
- Official repository for paper "Versatile Offline Imitation from Observations and Examples via Regularized State-Occupancy Matching" (ICML…☆29Jan 12, 2023Updated 3 years ago
- Official implementation of the Informed Dreamer algorithm, based on DreamerV3☆20Jan 29, 2026Updated last month
- Code release for the paper "Goal Representations for Instruction Following: A Semi-Supervised Language Interface to Control"☆17Apr 9, 2024Updated last year
- Code for FOCAL Paper Published at ICLR 2021☆55Dec 4, 2023Updated 2 years ago
- Online Decision Transformer☆275Jan 22, 2024Updated 2 years ago
- ☆14Mar 5, 2024Updated 2 years ago
- The official implementation of "Mind the Gap: Offline Policy Optimization for Imperfect Rewards" (ICLR2023)☆16Mar 3, 2023Updated 3 years ago
- ☆365May 1, 2023Updated 2 years ago
- ☆17Oct 12, 2023Updated 2 years ago
- ☆11Apr 10, 2023Updated 2 years ago
- Codes accompanying the paper "DOP: Off-Policy Multi-Agent Decomposed Policy Gradients" (ICLR 2021, https://arxiv.org/abs/2007.12322)☆51Dec 8, 2022Updated 3 years ago
- BASALT Benchmark datasets, evaluation code and agent training example.☆22Nov 29, 2023Updated 2 years ago
- ☆42May 11, 2022Updated 3 years ago
- Official implementation for "Anti-Exploration by Random Network Distillation", ICML 2023☆57Feb 3, 2023Updated 3 years ago
- Predictable MDP Abstraction for Unsupervised Model-Based RL (ICML 2023)☆32Feb 6, 2023Updated 3 years ago
- Simple behavioural cloning baseline solution for BASALT 2022☆33Nov 3, 2022Updated 3 years ago
- ☆15Apr 5, 2023Updated 2 years ago
- ☆16Aug 2, 2022Updated 3 years ago
- [ICML 2022] The official implementation of DWBC in "Discriminator-Weighted Offline Imitation Learning from Suboptimal Demonstrations"☆35Jan 5, 2023Updated 3 years ago
- ☆72Jun 20, 2022Updated 3 years ago
- Code for "Explore, Discover and Learn: Unsupervised Discovery of State-Covering Skills"☆34Feb 16, 2020Updated 6 years ago
- Codebase for ICLR 2023 paper, "SMART: Self-supervised Multi-task pretrAining with contRol Transformers"☆54Jan 26, 2024Updated 2 years ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Aug 8, 2022Updated 3 years ago
- Standalone library of frequently-used wrappers for dm_env environments.☆19Jul 9, 2024Updated last year
- ☆22Oct 4, 2021Updated 4 years ago
- Official implementation of Transformer Neural Processes☆82Aug 21, 2022Updated 3 years ago
- [NeurIPS 2022 Oral] The official implementation of POR in "A Policy-Guided Imitation Approach for Offline Reinforcement Learning"☆57Apr 6, 2023Updated 2 years ago