Code for "Masked Autoencoding for Scalable and Generalizable Decision Making". NeurIPS 2022
☆47Mar 12, 2024Updated last year
Alternatives and similar repositories for MaskDP_public
Users that are interested in MaskDP_public are comparing it to the libraries listed below
Sorting:
- MTM Masked Trajectory Models for Prediction, Representation, and Control.☆162Dec 16, 2025Updated 2 months ago
- A System for Morphology-Task Generalization via Unified Representation and Behavior Distillation (ICLR2023)☆14Feb 3, 2023Updated 3 years ago
- Codebase for "Uni[MASK]: Unified Inference in Sequential Decision Problems"☆57Jul 3, 2024Updated last year
- Code for Paper "Towards More Generalizable One-Shot Visual Imitation Learning", ICRA 2022☆20May 5, 2022Updated 3 years ago
- Chain-of-Thought Predictive Control☆57May 1, 2023Updated 2 years ago
- Reinforcement Learning with Latent Flow☆44Mar 25, 2021Updated 4 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
- Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)☆70Aug 8, 2022Updated 3 years ago
- ☆72Jun 20, 2022Updated 3 years ago
- Code of the paper "Universal Morphology Control via Contextual Modulation" at ICML 2023☆13Aug 3, 2023Updated 2 years ago
- Pessimistic Bootstrapping for Uncertainty-Driven Offline Reinforcement Learning☆29Feb 21, 2022Updated 4 years ago
- Implementation of Decision Stacks: Flexible RL via Modular Generative Models [NeurIPS 2023]☆12Jun 27, 2023Updated 2 years ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆129Feb 8, 2022Updated 4 years ago
- Implementation of "Open-World Multi-Task Control Through Goal-Aware Representation Learning and Adaptive Horizon Prediction"☆46Aug 15, 2023Updated 2 years ago
- Masked World Models for Visual Control☆135Jun 11, 2023Updated 2 years ago
- ☆360Oct 12, 2022Updated 3 years ago
- ☆16Jan 13, 2023Updated 3 years ago
- ☆14Dec 16, 2021Updated 4 years ago
- SpawnNet: Learning Generalizable Visuomotor Skills from Pre-trained Networks☆36Apr 29, 2024Updated last year
- Zero-Shot Learning in Named Entity Recognition with Common Sense Knowledge☆17Nov 16, 2021Updated 4 years ago
- Semantic-Aware Fine-Grained Correspondence, at ECCV 2022 (Oral)☆14Oct 29, 2022Updated 3 years ago
- A PyTorch Implementation of PlaNet: A Deep Planning Network for Reinforcement Learning☆12Aug 31, 2020Updated 5 years ago
- Code for paper "Keypoints into the Future: Self-Supervised Correspondence in Model-Based RL"☆13Nov 15, 2021Updated 4 years ago
- Generalised UDRL☆37May 12, 2022Updated 3 years ago
- ☆36May 24, 2023Updated 2 years ago
- ☆41Oct 16, 2023Updated 2 years ago
- ☆364May 1, 2023Updated 2 years ago
- SAPIEN Manipulation Skill Benchmark (NeurIPS 2021 Track on Datasets and Benchmarks)☆151Aug 24, 2022Updated 3 years ago
- The official implementation of "When Data Geometry Meets Deep Function: Generalizing Offline Reinforcement Learning" (ICLR2023)☆44Mar 6, 2023Updated 2 years ago
- HACMan: Learning Hybrid Actor-Critic Maps for 6D Non-Prehensile Manipulation☆48Oct 14, 2023Updated 2 years ago
- [NeurIPS 2021] PyTorch Code for Accelerating Robotic Reinforcement Learning with Parameterized Action Primitives☆79Jul 11, 2022Updated 3 years ago
- Original code for the paper "Learning Goal-Conditioned Policies Offline with Self-Supervised Reward Shaping" by Mezghani et al.☆18Jun 8, 2023Updated 2 years ago
- PyTorch implementation for all models and environments in the paper "Learning to Ground Multi-Agent Communication with Autoencoders"☆49Oct 29, 2021Updated 4 years ago
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆869Oct 14, 2024Updated last year
- We propose an evolution-based approach to meta-learn synthetic neural environments and reward neural networks for reinforcement learning.☆21Feb 23, 2023Updated 3 years ago
- Code for Abstract-to-Executable Trajectory Translation for One Shot Task Generalization (ICML 2023)☆23May 12, 2023Updated 2 years ago
- ☆18Jul 10, 2022Updated 3 years ago
- Annotated Tutorial for PerAct☆19Sep 11, 2023Updated 2 years ago
- TensorFlow implementation for our paper "Learning Long-Term Reward Redistribution via Randomized Return Decomposition"☆19Mar 17, 2022Updated 3 years ago