thuml / ContextWM
Code release for "Pre-training Contextualized World Models with In-the-wild Videos for Reinforcement Learning" (NeurIPS 2023), https://arxiv.org/abs/2305.18499
β55Updated last month
Related projects β
Alternatives and complementary repositories for ContextWM
- Official repository for "iVideoGPT: Interactive VideoGPTs are Scalable World Models" (NeurIPS 2024), https://arxiv.org/abs/2405.15223β70Updated 2 weeks ago
- [ECCV 2024] πOfficial implementation of the paper "Diffusion Reward: Learning Rewards via Conditional Video Diffusion"β78Updated 4 months ago
- β77Updated 3 months ago
- VP2 Benchmark (A Control-Centric Benchmark for Video Prediction, ICLR 2023)β23Updated 10 months ago
- [NeurIPS 2024] GenRL: Multimodal foundation world models allow grounding language and video prompts into embodied domains, by turning theβ¦β59Updated 3 months ago
- β46Updated 2 months ago
- Repository for "General Flow as Foundation Affordance for Scalable Robot Learning"β37Updated 7 months ago
- [ICML 2024] The offical Implementation of "DecisionNCE: Embodied Multimodal Representations via Implicit Preference Learning"β68Updated last month
- Official repository for "LIV: Language-Image Representations and Rewards for Robotic Control" (ICML 2023)β86Updated last year
- β68Updated 2 months ago
- Codebase for HiPβ87Updated 11 months ago
- Masked World Models for Visual Controlβ118Updated last year
- Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction οΌICML 2023οΌβ39Updated last year
- PWM: Policy Learning with Large World Modelsβ37Updated 3 months ago
- Code for the ICLR 2024 spotlight paper: "Learning to Act without Actions" (introducing Latent Action Policies)β76Updated 3 months ago
- Code for Reinforcement Learning from Vision Language Foundation Model Feedbackβ55Updated 5 months ago
- Code for paper "Grounding Video Models to Actions through Goal Conditioned Exploration".β24Updated last week
- β58Updated last month
- β45Updated 9 months ago
- β41Updated 10 months ago
- β42Updated 4 months ago
- β80Updated last week
- Code release for "HarmonyDream: Task Harmonization Inside World Models" (ICML 2024), https://arxiv.org/abs/2310.00344β26Updated 4 months ago
- β45Updated last month
- Implantation of CtrlFormerβ28Updated 2 years ago
- Official PyTorch implementation of AdaFlowβ27Updated last week
- Code for "Masked Autoencoding for Scalable and Generalizable Decision Making". NeurIPS 2022β44Updated 8 months ago
- Official code for "QueST: Self-Supervised Skill Abstractions for Continuous Control" [NeurIPS 2024]β38Updated last month
- DrM, a visual RL algorithm, minimizes the dormant ratio to guide exploration-exploitation trade-offs, achieving significant improvements β¦β62Updated 5 months ago
- Codebase of NeurIPS 2022 paper ''Planning for Sample Efficient Imitation Learning''β35Updated 2 years ago