Implementation of rectified flow and some of its followup research / improvements in Pytorch - View it on GitHub