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.

Consistency Models Made Easy

Official implementation of Decoupled MeanFlow

Official repo for CFG-Zero*

JAX implementation of MeanFlow

Pytorch implementation of MeanFlow on ImageNet and CIFAR10

Pytorch implementation for MeanFlow

Official implementation of Inductive Moment Matching

Pytorch Implementation (unofficial) of the paper "Mean Flows for One-step Generative Modeling" by Geng et al.

Official repo for consistency models.

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

⚡ InstaFlow! One-Step Stable Diffusion with Rectified Flow (ICLR 2024)

[ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)

The official code of "Weak-to-Strong Diffusion with Reflection".

PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

Official Implementation of Rectified Flow (ICLR2023 Spotlight)

Annotated Flow Matching paper

Free and Open Source, Distributed, RESTful Search Engine

Open-source vector similarity search for Postgres

[NeurIPS 2024] Simple and Effective Masked Diffusion Language Model

Reimplementation of Variational Inference with Normalizing Flows (https://arxiv.org/abs/1505.05770)

PyTorch implementation of normalizing flow models

The official Meta Llama 3 GitHub site

Official implementation of VQ-Diffusion

Vector (and Scalar) Quantization, in Pytorch

Implementation of Bit Diffusion, Hinton's group's attempt at discrete denoising diffusion, in Pytorch

A python module to scrape arxiv.org for a date range and category