The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024 - View it on GitHub