micahcarroll / uniMASK
Codebase for "Uni[MASK]: Unified Inference in Sequential Decision Problems"
☆54Updated 7 months ago
Alternatives and similar repositories for uniMASK:
Users that are interested in uniMASK are comparing it to the libraries listed below
- Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)☆67Updated 2 years ago
- ☆42Updated last year
- ☆47Updated last year
- Code for "Masked Autoencoding for Scalable and Generalizable Decision Making". NeurIPS 2022☆43Updated 11 months ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆76Updated 2 months ago
- Reinforcement Learning via Supervised Learning☆70Updated 2 years ago
- ☆29Updated 2 years ago
- ☆40Updated 3 years ago
- Change-Based Exploration Transfer☆36Updated 2 years ago
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Updated 9 months ago
- Conservative Q learning in Jax☆52Updated 2 years ago
- ☆54Updated 11 months ago
- Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023☆28Updated last year
- Official code repository for Prompt-DT.☆102Updated 2 years ago
- CaDM: Context-aware Dynamics Model for Generalization in Model-based Reinforcement Learning☆62Updated 4 years ago
- Official Codebase for Offline Reinforcement Learning from Images with Latent Space Models☆31Updated 3 years ago
- Pre-Training Goal-based Models for Sample-Efficient Reinforcement Learning☆25Updated 11 months ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆109Updated 3 years ago
- Official code for "Unleashing the Power of Pre-trained Language Models for Offline Reinforcement Learning".☆39Updated 10 months ago
- [ICLR 2023 Oral] The official implementation of SQL and EQL in "Offline RL with No OOD Actions: In-Sample Learning via Implicit Value Reg…☆44Updated last year
- Reward shaping approach for instruction following settings, leveraging language at multiple levels of abstraction.☆19Updated 3 years ago
- Clean, extensible implementation of MACAW [ICML 2021]☆13Updated 3 years ago
- Learning to Modulate pre-trained Models in RL (Decision Transformer, LoRA, Fine-tuning)☆52Updated 4 months ago
- Implemention of the Decision-Pretrained Transformer (DPT) from the paper Supervised Pretraining Can Learn In-Context Reinforcement Learni…☆57Updated 8 months ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆29Updated 6 months ago
- My Body Is A Cage☆39Updated 3 years ago
- ☆41Updated 3 years ago
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆60Updated last year
- Official code for "Pretraining Representations For Data-Efficient Reinforcement Learning" (NeurIPS 2021)☆52Updated 3 years ago
- Author's PyTorch implementation of SR-DICE for marginalized importance sampling☆15Updated 3 years ago