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
☆54Updated 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☆68Updated last week
- [NeurIPS 2024] GenRL: Multimodal foundation world models allow grounding language and video prompts into embodied domains, by turning the…☆58Updated 3 months ago
- [ECCV 2024] 💐Official implementation of the paper "Diffusion Reward: Learning Rewards via Conditional Video Diffusion"☆76Updated 4 months ago
- [ICML 2024] The offical Implementation of "DecisionNCE: Embodied Multimodal Representations via Implicit Preference Learning"☆67Updated last month
- ☆76Updated 2 months ago
- Codebase for HiP☆86Updated 10 months ago
- Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction (ICML 2023)☆39Updated last year
- VP2 Benchmark (A Control-Centric Benchmark for Video Prediction, ICLR 2023)☆23Updated 10 months ago
- ☆42Updated 2 months ago
- Repository for "General Flow as Foundation Affordance for Scalable Robot Learning"☆37Updated 6 months ago
- Masked World Models for Visual Control☆118Updated last year
- ☆68Updated 2 months ago
- Official repository for "LIV: Language-Image Representations and Rewards for Robotic Control" (ICML 2023)☆84Updated last year
- ☆57Updated 3 weeks ago
- PWM: Policy Learning with Large World Models☆37Updated 2 months ago
- [NeurIPS 2023] Efficient Diffusion Policy☆81Updated last year
- ☆64Updated 2 weeks ago
- Pytorch implementation of "Genie: Generative Interactive Environments", Bruce et al. (2024).☆68Updated 2 months ago
- ☆45Updated 9 months ago
- Implantation of CtrlFormer☆28Updated 2 years ago
- ☆21Updated 8 months ago
- ☆29Updated 2 weeks ago
- Code for the ICLR 2024 spotlight paper: "Learning to Act without Actions" (introducing Latent Action Policies)☆74Updated 3 months ago
- Code for "Masked Autoencoding for Scalable and Generalizable Decision Making". NeurIPS 2022☆43Updated 7 months ago
- ☆41Updated 4 months ago
- Official PyTorch implementation of AdaFlow☆19Updated 5 months ago
- ☆41Updated 9 months ago
- ☆52Updated last year
- [RA-L 2023 & IROS 2023] Visual Reinforcement Learning with Self-Supervised 3D Representations☆76Updated last year
- [ICRA2023] Grounding Language with Visual Affordances over Unstructured Data☆36Updated last year