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
β58Updated 4 months ago
Alternatives and similar repositories for ContextWM:
Users that are interested in ContextWM are comparing it to the libraries listed below
- β59Updated 4 months ago
- [ECCV 2024] πOfficial implementation of the paper "Diffusion Reward: Learning Rewards via Conditional Video Diffusion"β88Updated 6 months ago
- [NeurIPS 2024] GenRL: Multimodal-foundation world models enable grounding language and video prompts into embodied domains, by turning thβ¦β63Updated 2 weeks ago
- VP2 Benchmark (A Control-Centric Benchmark for Video Prediction, ICLR 2023)β25Updated last year
- [ICML 2024] The offical Implementation of "DecisionNCE: Embodied Multimodal Representations via Implicit Preference Learning"β71Updated 4 months ago
- Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction οΌICML 2023οΌβ45Updated last year
- β88Updated 5 months ago
- β73Updated 5 months ago
- Official repository for "iVideoGPT: Interactive VideoGPTs are Scalable World Models" (NeurIPS 2024), https://arxiv.org/abs/2405.15223β104Updated last week
- Codebase for HiPβ88Updated last year
- β47Updated 7 months ago
- Code for paper "Grounding Video Models to Actions through Goal Conditioned Exploration".β40Updated last month
- β20Updated this week
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"β51Updated this week
- Official code for "QueST: Self-Supervised Skill Abstractions for Continuous Control" [NeurIPS 2024]β56Updated 2 months ago
- PWM: Policy Learning with Large World Modelsβ39Updated 5 months ago
- Code for the ICLR 2024 spotlight paper: "Learning to Act without Actions" (introducing Latent Action Policies)β84Updated 5 months ago
- Implantation of CtrlFormerβ28Updated 2 years ago
- An ML research template with good documentation by Boyuan Chen, an MIT PhD studentβ28Updated last month
- Uni-RLHF platform for "Uni-RLHF: Universal Platform and Benchmark Suite for Reinforcement Learning with Diverse Human Feedback" (ICLR2024β¦β33Updated 2 months ago
- Source codes for the paper "COMBO: Compositional World Models for Embodied Multi-Agent Cooperation"β28Updated 9 months ago
- Masked World Models for Visual Controlβ120Updated last year
- Official repository for "LIV: Language-Image Representations and Rewards for Robotic Control" (ICML 2023)β96Updated last year
- Code for Stable Control Representationsβ23Updated 3 weeks ago
- code for the paper Imitation Learning from Observation with Automatic Discount Schedulingβ13Updated 10 months ago
- Codes accompanying the paper "Score Regularized Policy Optimization through Diffusion Behavior" (ICLR 2024).β42Updated 11 months ago
- [NeurIPS 2023] Efficient Diffusion Policyβ91Updated last year
- GRAPE: Guided-Reinforced Vision-Language-Action Preference Optimizationβ69Updated last month
- Official PyTorch implementation of AdaFlowβ45Updated 2 months ago