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,153Updated 2 months ago
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☆1,877Updated 2 weeks ago
- PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838☆1,686Updated 10 months ago
- PyTorch implementation of FractalGen https://arxiv.org/abs/2502.17437☆1,152Updated 5 months ago
- Official Implementation of Rectified Flow (ICLR2023 Spotlight)☆1,360Updated last year
- Interactive visualizations of the geometric intuition behind diffusion models.☆806Updated last month
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,743Updated last year
- Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI☆1,184Updated last month
- A summary of related works about flow matching, stochastic interpolants☆525Updated 4 months ago
- PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆1,967Updated last year
- Vector (and Scalar) Quantization, in Pytorch☆3,451Updated 2 weeks ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆948Updated 4 months ago
- Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆1,702Updated 2 years ago
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆1,232Updated 4 months ago
- ☆592Updated 8 months ago
- Simple and readable code for training and sampling from diffusion models☆573Updated last month
- EDM2 and Autoguidance -- Official PyTorch implementation☆746Updated 7 months ago
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆928Updated last year
- A suite of image and video neural tokenizers☆1,659Updated 5 months ago
- Code for NeurIPS 2024 paper - The GAN is dead; long live the GAN! A Modern Baseline GAN - by Huang et al.☆789Updated 6 months ago
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆924Updated last year
- Autoregressive Model Beats Diffusion: 🦙 Llama for Scalable Image Generation☆1,821Updated 11 months ago
- Muon is an optimizer for hidden layers in neural networks☆1,390Updated 3 weeks ago
- [CVPR 2025 Oral] Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models☆1,077Updated last month
- This repo contains the code for 1D tokenizer and generator☆975Updated 4 months ago
- Schedule-Free Optimization in PyTorch☆2,196Updated 2 months ago
- Pytorch Implementation (unofficial) of the paper "Mean Flows for One-step Generative Modeling" by Geng et al.☆682Updated last month
- Denoising Diffusion Implicit Models☆1,679Updated last year
- A curated list for awesome discrete diffusion models resources.☆416Updated 2 months ago
- A series of tutorial notebooks on denoising diffusion probabilistic models in PyTorch☆696Updated 2 years ago
- [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)☆610Updated last year