XiangchengZhang / Diffusion-inference-scalingLinks
Official Implementation for Inference-time Scaling of Diffusion Models through Classical Search
☆25Updated 2 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:
- ☆79Updated last year
- The official implementations of Intention-conditioned Flow Occupancy Models (InFOM)☆30Updated 2 months ago
- Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction (ICML 2023)☆50Updated 2 years ago
- Source codes for the paper "COMBO: Compositional World Models for Embodied Multi-Agent Cooperation"☆43Updated 9 months ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆97Updated 10 months ago
- ☆72Updated 6 months ago
- ☆50Updated last month
- Official Codebase for "Aligning Diffusion Behaviors with Q-functions for Efficient Continuous Control" (NeurIPS 2024)☆15Updated last year
- [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"☆28Updated 6 months ago
- A framework for training world models with virtual environments, complete with annotated environment dataset (RetroAct), exploration agen…☆66Updated last month
- Code for the tutorial/review paper for RL-based-fine-tuniing. In this code, we especially focus on the design of biological sequences li…☆155Updated last year
- The official code release for Q#: Provably Optimal Distributional RL for LLM Post-Training☆17Updated 9 months ago
- [NeurIPS 2024] GenRL: Multimodal-foundation world models enable grounding language and video prompts into embodied domains, by turning th…☆85Updated 8 months ago
- Official implementation for: Consistency Models as a Rich and Efficient Policy Class for Reinforcement Learning ICLR'24☆26Updated last year
- Official implementation of the RLC 2024 paper "Policy-Guided Diffusion"☆148Updated last year
- The official implementation of flow Q-learning (FQL)☆261Updated 4 months ago
- SEIKO is a novel reinforcement learning method to efficiently fine-tune diffusion models in an online setting. Our methods outperform all…☆29Updated last year
- Public code for "Reinforcement Learning from Passive Data via Latent Intentions"☆89Updated 2 years ago
- ☆170Updated 3 weeks ago
- Code release for "Pre-training Contextualized World Models with In-the-wild Videos for Reinforcement Learning" (NeurIPS 2023), https://ar…☆69Updated last year
- Pytorch implementation of "Genie: Generative Interactive Environments", Bruce et al. (2024).☆238Updated last year
- ☆40Updated last year
- ☆25Updated last year
- PWM: Policy Learning with Large World Models☆61Updated 4 months ago
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆77Updated last year
- Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction☆35Updated 2 years ago
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆57Updated 6 months ago
- We develop world models that can be adapted with natural language. Intergrating these models into artificial agents allows humans to effe…☆25Updated last year
- ☆78Updated 6 months ago
- [ICML'25] The PyTorch implementation of paper: "AdaWorld: Learning Adaptable World Models with Latent Actions".☆183Updated 6 months ago