raywang4 / EqMLinks
☆158Updated last month
Alternatives and similar repositories for EqM
Users that are interested in EqM are comparing it to the libraries listed below
Sorting:
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆194Updated 4 months ago
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆124Updated last year
- [ICCV 2025] Official Implementation of Contrastive Flow Matching☆147Updated 4 months ago
- [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)☆159Updated last week
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆51Updated 5 months ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆139Updated 8 months ago
- Educational implementation of the Discrete Flow Matching paper☆126Updated last year
- ☆158Updated last month
- Inference-Time Alignment in Protein Diffusion Models☆47Updated 9 months ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆247Updated 10 months ago
- ☆296Updated 11 months ago
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆158Updated 9 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆394Updated 3 months ago
- My take on Flow Matching☆84Updated 10 months ago
- ☆73Updated 3 months ago
- Official Github Repo for Neurips 2024 Paper Immiscible Diffusion: Accelerating Diffusion Training with Noise Assignment☆61Updated 5 months ago
- Annotated Flow Matching paper☆218Updated last year
- [ECCV 2024, Oral] FMBoost: Boosting Latent Diffusion with Flow Matching☆249Updated last month
- ☆62Updated 3 months ago
- The aim of this repository is to test and implement Flow-Matching-based models☆118Updated 10 months ago
- Flash Attention Triton kernel with support for second-order derivatives☆111Updated 3 weeks ago
- ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC☆146Updated last year
- Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]☆82Updated 6 months ago
- The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)☆98Updated 10 months ago
- [NeurIPS 2025] Official implementation for our paper "Scaling Diffusion Transformers Efficiently via μP".☆92Updated 2 weeks ago
- Code for the paper DisCo-Diff: Enhancing Continuous Diffusion Models with Discrete Latents, ICML 2024☆91Updated last year
- [NeurIPS 2025] Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling"☆175Updated last month
- [Preprint] UCGM: Unified Continuous Generative Models☆169Updated 5 months ago
- Code for TFG: Unified Training-Free Guidance for Diffusion Models☆70Updated 2 weeks ago
- Simple Guidance Mechanisms for Discrete Diffusion Models☆58Updated 11 months ago