zacharyhorvitz / Fk-Diffusion-SteeringView external linksLinks
A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.
☆211Jun 26, 2025Updated 7 months ago
Alternatives and similar repositories for Fk-Diffusion-Steering
Users that are interested in Fk-Diffusion-Steering are comparing it to the libraries listed below
Sorting:
- Inference-time scaling of diffusion-based image and video generation models.☆172Dec 17, 2025Updated last month
- Inference-Time Alignment in Protein Diffusion Models☆52Jan 7, 2026Updated last month
- Official Implementation of the paper: A Complete Recipe for Diffusion Generative Models☆31Nov 1, 2024Updated last year
- [CVPR 2025] Aesthetic Post-Training Diffusion Models from Generic Preferences with Step-by-step Preference Optimization☆262Apr 7, 2025Updated 10 months ago
- Official implementation for Diffusion Alignment as Sampling (DAS), ICLR'25, Spotlight☆59Feb 12, 2025Updated last year
- ☆12Nov 22, 2024Updated last year
- EDM2 and Autoguidance -- Official PyTorch implementation☆819Dec 9, 2024Updated last year
- [NeurIPS 2025] Official code for Inference-Time Scaling for Flow Models via Stochastic Generation and Rollover Budget Forcing☆72Oct 12, 2025Updated 4 months ago
- Reflect-DiT: Inference-Time Scaling for Text-to-Image Diffusion Transformers via In-Context Reflection☆55Aug 16, 2025Updated 6 months ago
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆4,107Jan 5, 2026Updated last month
- Maximize the Resolution Potential of Pre-trained Rectified Flow Transformers☆65Oct 16, 2024Updated last year
- Dirichlet Diffusion Score Model for Biological Sequence Generation.☆64Feb 12, 2025Updated last year
- [CVPR2025] PyTorch-based reimplementation of CrossFlow, as proposed in 'Flowing from Words to Pixels: A Noise-Free Framework for Cross-Mo…☆330Jun 8, 2025Updated 8 months ago
- ☆13Jan 22, 2025Updated last year
- Code for "PartSDF: Part-Based Implicit Neural Representation for Composite 3D Shape Parametrization and Optimization"☆20Nov 4, 2025Updated 3 months ago
- Notebooks to demonstrate TimmWrapper☆16Jan 16, 2025Updated last year
- Code for "Aligning Optimization Trajectories with Diffusion Models for Constrained Design Generation" @ NeurIPS 2023☆24Oct 12, 2023Updated 2 years ago
- ☆206Feb 12, 2024Updated 2 years ago
- Derivative-Free Guidance in Diffusion Models with Soft Value-Based Decoding. For controlled generation in DNA, RNA, proteins, molecules (…☆36Jan 7, 2026Updated last month
- ☆15Sep 22, 2025Updated 4 months ago
- Code for the paper https://arxiv.org/abs/2402.04997☆104Feb 8, 2024Updated 2 years ago
- SEIKO is a novel reinforcement learning method to efficiently fine-tune diffusion models in an online setting. Our methods outperform all…☆29Jul 18, 2024Updated last year
- Diffusion-Sharpening: Fine-tuning Diffusion Models with Denoising Trajectory Sharpening☆69May 18, 2025Updated 8 months ago
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆245Mar 11, 2025Updated 11 months ago
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆1,544Mar 16, 2025Updated 10 months ago
- Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist☆64Dec 16, 2023Updated 2 years ago
- Proteina is a new large-scale flow-based protein backbone generator that utilizes hierarchical fold class labels for conditioning and rel…☆234Jul 24, 2025Updated 6 months ago
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆164Jan 31, 2025Updated last year
- Official implementation of Deep Momentum Schrödinger Bridge☆30Nov 5, 2023Updated 2 years ago
- [ACM Computing Surveys] The collection of awesome papers on alignment of diffusion models.☆403Feb 6, 2026Updated last week
- ☆739Dec 5, 2024Updated last year
- [NeurIPS 2024] EvolveDirector: Approaching Advanced Text-to-Image Generation with Large Vision-Language Models.☆51Oct 14, 2024Updated last year
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,903Mar 16, 2024Updated last year
- FoldFlow: SE(3)-Stochastic Flow Matching for Protein Backbone Generation☆274Dec 10, 2024Updated last year
- PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Accelerator (NeurIPS 2024)☆534Sep 8, 2025Updated 5 months ago
- Educational implementation of the Discrete Flow Matching paper☆132Aug 26, 2024Updated last year
- Consistency Models Made Easy☆324Oct 13, 2024Updated last year
- PrismLayers: Open Data for High-Quality Multi-Layer Transparent Image Generative Models☆34Jan 14, 2026Updated last month
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆1,094Dec 22, 2025Updated last month