XiangchengZhang / Diffusion-inference-scalingLinks
Official Implementation for Inference-time Scaling of Diffusion Models through Classical Search
☆22Updated this week
Alternatives and similar repositories for Diffusion-inference-scaling
Users that are interested in Diffusion-inference-scaling are comparing it to the libraries listed below
Sorting:
- ☆73Updated last year
- Pytorch implementation of "Genie: Generative Interactive Environments", Bruce et al. (2024).☆208Updated last year
- Implementations of Intention-conditioned Flow Occupancy Models (InFOM)☆28Updated last week
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆83Updated 8 months ago
- The official implementation of flow Q-learning (FQL)☆236Updated 2 months ago
- Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction (ICML 2023 )☆50Updated 2 years ago
- ☆50Updated 4 months ago
- A framework for training world models with virtual environments, complete with annotated environment dataset (RetroAct), exploration agen…☆60Updated last week
- Source codes for the paper "COMBO: Compositional World Models for Embodied Multi-Agent Cooperation"☆42Updated 7 months ago
- Official implementation of the RLC 2024 paper "Policy-Guided Diffusion"☆143Updated last year
- ☆20Updated 2 months ago
- Codebase for HiP☆90Updated last year
- ☆42Updated 3 months ago
- An ML research template with good documentation by Boyuan Chen, an MIT PhD student☆86Updated 7 months ago
- Official implementation of "Flow Based Policy for Online Reinforcement Learning"☆41Updated last month
- [ICML'25] The PyTorch implementation of paper: "AdaWorld: Learning Adaptable World Models with Latent Actions".☆161Updated 3 months ago
- [NeurIPS 2024] GenRL: Multimodal-foundation world models enable grounding language and video prompts into embodied domains, by turning th…☆81Updated 6 months ago
- [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"☆28Updated 4 months ago
- Official repository for "iVideoGPT: Interactive VideoGPTs are Scalable World Models" (NeurIPS 2024), https://arxiv.org/abs/2405.15223☆153Updated 3 weeks ago
- [ICML 2024] LLM and Simulation as Bilevel Optimizers: A New Paradigm to Advance Physical Scientific Discovery☆79Updated last year
- Code release for "Pre-training Contextualized World Models with In-the-wild Videos for Reinforcement Learning" (NeurIPS 2023), https://ar…☆67Updated last year
- Official implementation of paper "ROCKET-1: Mastering Open-World Interaction with Visual-Temporal Context Prompting" (CVPR 2025)☆45Updated 6 months ago
- ☆77Updated 4 months ago
- Using advances in generative modeling to learn reward functions from unlabeled videos.☆135Updated last year
- The official repository for NeurIPS 2024 Oral <Maximum Entropy Inverse Reinforcement Learning of Diffusion Models with Energy-Based Model…☆29Updated 6 months ago
- Official repository for "Investigating Pre-Training Objectives for Generalization in Visual Reinforcement Learning" (ICML 2024)☆11Updated 3 weeks ago
- GROOT: Learning to Follow Instructions by Watching Gameplay Videos (ICLR 2024 Spotlight)☆66Updated last year
- Official repository for "RLVR-World: Training World Models with Reinforcement Learning" (NeurIPS 2025), https://arxiv.org/abs/2505.13934☆108Updated 2 weeks ago
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆137Updated 5 months ago
- [ICML 2025 Oral] Official repo of EmbodiedBench, a comprehensive benchmark designed to evaluate MLLMs as embodied agents.☆194Updated 2 months ago