facebookresearch / flow_matchingLinks
A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes practical examples for both text and image modalities.
☆3,707Updated last month
Alternatives and similar repositories for flow_matching
Users that are interested in flow_matching are comparing it to the libraries listed below
Sorting:
- TorchCFM: a Conditional Flow Matching library☆2,110Updated last week
- PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838☆1,786Updated last year
- Official Implementation of Rectified Flow (ICLR2023 Spotlight)☆1,461Updated last year
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,824Updated last year
- PyTorch implementation of FractalGen https://arxiv.org/abs/2502.17437☆1,193Updated 8 months ago
- PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆2,033Updated last year
- Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI☆1,259Updated last month
- A summary of related works about flow matching, stochastic interpolants☆596Updated 7 months ago
- Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆1,755Updated 2 years ago
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆1,412Updated 8 months ago
- Vector (and Scalar) Quantization, in Pytorch☆3,689Updated last week
- Interactive visualizations of the geometric intuition behind diffusion models.☆897Updated 5 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆1,073Updated last week
- A suite of image and video neural tokenizers☆1,679Updated 9 months ago
- ☆681Updated 11 months ago
- Pytorch Implementation (unofficial) of the paper "Mean Flows for One-step Generative Modeling" by Geng et al.☆928Updated last month
- EDM2 and Autoguidance -- Official PyTorch implementation☆790Updated 11 months ago
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆1,021Updated 2 weeks ago
- [CVPR 2025 Oral] Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models☆1,285Updated 5 months ago
- Denoising Diffusion Implicit Models☆1,738Updated last year
- Schedule-Free Optimization in PyTorch☆2,229Updated 5 months ago
- Simple and readable code for training and sampling from diffusion models☆658Updated 5 months ago
- Code for NeurIPS 2024 paper - The GAN is dead; long live the GAN! A Modern Baseline GAN - by Huang et al.☆828Updated 9 months ago
- A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)☆837Updated 2 years ago
- Muon is an optimizer for hidden layers in neural networks☆2,002Updated 4 months ago
- This repo contains the code for 1D tokenizer and generator☆1,071Updated 7 months ago
- Pytorch implementation of Diffusion Models (https://arxiv.org/pdf/2006.11239.pdf)☆1,390Updated 2 years ago
- Denoising Diffusion Probabilistic Models☆4,837Updated 2 years ago
- Autoregressive Model Beats Diffusion: 🦙 Llama for Scalable Image Generation☆1,889Updated last year
- Conditional diffusion model to generate MNIST. Minimal script. Based on 'Classifier-Free Diffusion Guidance'.☆810Updated last year