This is a PyTorch implementation of the paper "Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flow". It includes core algorithm and visualisations using data points for better understanding.
It is accompanied by a Notion blog that documents my learning - Blog
Hope it helps others as well.