lucidrains/rectified-flow-pytorch
Implementation of rectified flow and some of its followup research / improvements in Pytorch
This tool helps researchers and practitioners in machine learning to train and use Rectified Flow models, which are advanced generative models. You provide a collection of existing images, and the system learns to generate new, similar images. It's designed for machine learning researchers, deep learning engineers, and data scientists working on generative AI.
426 stars and 1,635 monthly downloads. Available on PyPI.
Use this if you are developing or experimenting with generative AI models and want to leverage or improve upon Rectified Flow for high-quality image synthesis.
Not ideal if you are looking for a pre-trained, ready-to-use image generation application without needing to understand or implement the underlying model training.
Stars
426
Forks
30
Language
Python
License
MIT
Category
Last pushed
Mar 26, 2026
Monthly downloads
1,635
Commits (30d)
0
Dependencies
12
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/lucidrains/rectified-flow-pytorch"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related frameworks
probabilists/zuko
Normalizing flows in PyTorch
davidnabergoj/torchflows
Modern normalizing flows in Python. Simple to use and easily extensible.
keishihara/flow-matching
Flow Matching implemented in PyTorch
LukasRinder/normalizing-flows
Implementation of normalizing flows in TensorFlow 2 including a small tutorial.
janosh/awesome-normalizing-flows
Awesome resources on normalizing flows.