XiangchengZhang / Diffusion-inference-scalingLinks
Official Implementation for Inference-time Scaling of Diffusion Models through Classical Search
☆27Updated 3 months ago
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:
- ☆82Updated last year
- The official implementations of Intention-conditioned Flow Occupancy Models (InFOM)☆29Updated last week
- Source codes for the paper "COMBO: Compositional World Models for Embodied Multi-Agent Cooperation"☆45Updated 10 months ago
- Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction (ICML 2023)☆50Updated 2 years ago
- A framework for training world models with virtual environments, complete with annotated environment dataset (RetroAct), exploration agen…☆68Updated 2 months ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆97Updated 11 months ago
- Pytorch implementation of "Genie: Generative Interactive Environments", Bruce et al. (2024).☆246Updated last year
- Official Codebase for "Aligning Diffusion Behaviors with Q-functions for Efficient Continuous Control" (NeurIPS 2024)☆15Updated last year
- ☆51Updated 2 months ago
- Official implementation of paper "ROCKET-1: Mastering Open-World Interaction with Visual-Temporal Context Prompting" (CVPR'25)☆46Updated 9 months ago
- maze datasets for investigating OOD behavior of ML systems☆70Updated 2 months ago
- The official implementation of flow Q-learning (FQL)☆270Updated 5 months ago
- Selected list of papers on World Models that I found interesting and/or useful.☆31Updated 11 months ago
- Code for the tutorial/review paper for RL-based-fine-tuniing. In this code, we especially focus on the design of biological sequences li…☆156Updated last year
- Official repository for "RLVR-World: Training World Models with Reinforcement Learning" (NeurIPS 2025), https://arxiv.org/abs/2505.13934☆181Updated 2 months ago
- ☆75Updated 7 months ago
- Code release for "Pre-training Contextualized World Models with In-the-wild Videos for Reinforcement Learning" (NeurIPS 2023), https://ar…☆69Updated last year
- ☆231Updated last month
- [ICLR 2025] Official implementation and benchmark evaluation repository of <PhysBench: Benchmarking and Enhancing Vision-Language Models …☆83Updated 7 months ago
- IMG: Calibrating Diffusion Models via Implicit Multimodal Guidance, ICCV 2025☆30Updated 3 months ago
- Official repository for "iVideoGPT: Interactive VideoGPTs are Scalable World Models" (NeurIPS 2024), https://arxiv.org/abs/2405.15223☆162Updated 3 months ago
- [ICML 2024] LLM and Simulation as Bilevel Optimizers: A New Paradigm to Advance Physical Scientific Discovery☆81Updated last year
- [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"☆28Updated 7 months ago
- [IROS'25 Oral & NeurIPSw'24] Official implementation of "MineDreamer: Learning to Follow Instructions via Chain-of-Imagination for Simula…☆99Updated 6 months ago
- Paper collections of the continuous effort start from World Models.☆196Updated last year
- Official implementation of the RLC 2024 paper "Policy-Guided Diffusion"☆149Updated last year
- An ML research template with good documentation by Boyuan Chen, an MIT PhD student☆105Updated 10 months ago
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆79Updated last year
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Updated last year
- [ICML 2025] The official implementation of the paper "On the Guidance of Flow Matching"☆74Updated 3 months ago