facebookresearch / flow_matching
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.
☆2,206Updated 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
- TorchCFM: a Conditional Flow Matching library☆1,560Updated last week
- PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838☆1,371Updated 5 months ago
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,573Updated last year
- EDM2 and Autoguidance -- Official PyTorch implementation☆663Updated 3 months ago
- Official Implementation of Rectified Flow (ICLR2023 Spotlight)☆1,164Updated 8 months ago
- Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI☆982Updated this week
- A summary of related works about flow matching, stochastic interpolants☆407Updated 7 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆777Updated last week
- Schedule-Free Optimization in PyTorch☆2,116Updated 3 weeks ago
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆882Updated last week
- Simple and readable code for training and sampling from diffusion models☆445Updated 2 months ago
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆715Updated 10 months ago
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆790Updated last year
- Code for NeurIPS 2024 paper - The GAN is dead; long live the GAN! A Modern Baseline GAN - by Huang et al.☆702Updated 2 months ago
- PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆1,869Updated 8 months ago
- [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)☆530Updated last year
- Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆1,596Updated 2 years ago
- ☆411Updated 3 months ago
- SEED-Voken: A Series of Powerful Visual Tokenizers☆844Updated last month
- Autoregressive Model Beats Diffusion: 🦙 Llama for Scalable Image Generation☆1,620Updated 7 months ago
- Vector (and Scalar) Quantization, in Pytorch☆3,043Updated this week
- This repo contains the code for 1D tokenizer and generator☆748Updated last week
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆565Updated 3 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆364Updated 4 months ago
- A curated list for awesome discrete diffusion models resources.☆268Updated last month
- Conditional diffusion model to generate MNIST. Minimal script. Based on 'Classifier-Free Diffusion Guidance'.☆736Updated last year
- A series of tutorial notebooks on denoising diffusion probabilistic models in PyTorch☆670Updated 2 years ago
- Official repository for "AM-RADIO: Reduce All Domains Into One"☆947Updated last week
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,356Updated 2 months ago