Tutorial jupyter notebook to learn how to implement flow matching in PyTorch with solutions. In the end you will train a small model to generate mnist digits!
If you want to learn by yourself, feel free to complete the skeleton notebook (coming soon). If you are ever stuck, don't hesitate to look at my solutions or the papers referred at the beginning/end of the notebook.
Here are the mnist digits I get for the number 9:
If you obtain better results, you can let me know or PR into this repo.