lucidrains / rectified-flow-pytorchView external linksLinks
Implementation of rectified flow and some of its followup research / improvements in Pytorch
☆427Jan 30, 2026Updated 2 weeks ago
Alternatives and similar repositories for rectified-flow-pytorch
Users that are interested in rectified-flow-pytorch are comparing it to the libraries listed below
Sorting:
- Implementation of Autoregressive Diffusion in Pytorch☆432Dec 4, 2025Updated 2 months ago
- Official Implementation of Rectified Flow (ICLR2023 Spotlight)☆1,543Jul 20, 2024Updated last year
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆128Oct 18, 2024Updated last year
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆514Jan 18, 2026Updated 3 weeks ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆258Jan 17, 2025Updated last year
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆632Jul 1, 2024Updated last year
- TorchCFM: a Conditional Flow Matching library☆2,294Nov 11, 2025Updated 3 months ago
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆4,107Jan 5, 2026Updated last month
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆1,094Dec 22, 2025Updated last month
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆245Mar 11, 2025Updated 11 months ago
- A summary of related works about flow matching, stochastic interpolants☆636Feb 4, 2026Updated last week
- [ECCV 2024, Oral] FMBoost: Boosting Latent Diffusion with Flow Matching☆256Oct 17, 2025Updated 3 months ago
- The aim of this repository is to test and implement Flow-Matching-based models☆123Jan 15, 2025Updated last year
- EDM2 and Autoguidance -- Official PyTorch implementation☆819Dec 9, 2024Updated last year
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆1,544Mar 16, 2025Updated 10 months ago
- Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI☆1,326Jan 27, 2026Updated 2 weeks ago
- Official Implementation (Pytorch) of "Constant Acceleration Flow", NeurIPS 2024☆35Oct 31, 2025Updated 3 months ago
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆182Jun 20, 2024Updated last year
- Consistency Models Made Easy☆324Oct 13, 2024Updated last year
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆135Oct 15, 2025Updated 4 months ago
- InstaFlow! One-Step Stable Diffusion with Rectified Flow (ICLR 2024)☆1,386Jun 7, 2024Updated last year
- PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838☆1,859Sep 27, 2024Updated last year
- MeanAudio: Fast and Faithful Text-to-Audio Generation with Mean Flows☆123Sep 2, 2025Updated 5 months ago
- [ICASSP 2024] This is the official code for "VoiceFlow: Efficient Text-to-Speech with Rectified Flow Matching"☆366Sep 3, 2024Updated last year
- A mini-library for training consistency models.☆251Dec 26, 2023Updated 2 years ago
- Vector (and Scalar) Quantization, in Pytorch☆3,870Updated this week
- ☆82Jan 22, 2025Updated last year
- A family of state-of-the-art Transformer-based audio codecs for low-bitrate high-quality audio coding.☆416Sep 15, 2025Updated 5 months ago
- Variable Bitrate Residual Vector Quantization for Audio Coding☆51May 1, 2025Updated 9 months ago
- Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"☆8,352May 31, 2024Updated last year
- Make-An-Audio-3: Transforming Text/Video into Audio via Flow-based Large Diffusion Transformers☆118May 19, 2025Updated 8 months ago
- ☆10Sep 21, 2024Updated last year
- PyTorch implementation of the ICASSP-24 paper: "Improving Audio Captioning Models with Fine-grained Audio Features, Text Embedding Superv…☆38Jan 6, 2024Updated 2 years ago
- Contains the code associated with the ICLR submission for our text-to-speech diffusion model☆57Oct 31, 2023Updated 2 years ago
- ☆320Dec 17, 2024Updated last year
- [NAACL 2025] WaveFM: A High-Fidelity and Efficient Vocoder Based on Flow Matching☆121Mar 27, 2025Updated 10 months ago
- Official PyTorch implementation of the paper: Flow Matching in Latent Space☆334Jan 20, 2025Updated last year
- Pytorch Implementation (unofficial) of the paper "Mean Flows for One-step Generative Modeling" by Geng et al.☆1,070Dec 17, 2025Updated last month
- Official code for paper: Text-to-Image Rectified Flow as Plug-and-Play Priors [ICLR 2025]☆140Apr 16, 2025Updated 10 months ago