This algorithm comes from "Improving the Training of Rectified Flows" <img width="1362" height="473" alt="Image" src="https://github.com/user-attachments/assets/1c8bb57c-4a0e-4676-9072-4f30569e7b9a" />