Flow Matching for Generative Modeling

notebook by Georges Le Bellier - Twitter, Website

⚠️ This is not the official implementation of the original paper.

This notebook centers around the Flow Matching for Generative Modeling article [1] and proposes an implementation of Flow Matching in the case of Optimal Transport conditional Vector Fields. The implementation proposed in [2] was consulted and it inspired the use of the Zuko [3] package for ODE solving and sampling. Moreover, this notebook adopts the notations of the original article and thus the numbers of the equations are the same as in the paper.

References:

📄 [1] Flow Matching for Generative Modeling by Yaron Lipman, Ricky T. Q. Chen, Heli Ben-Hamu, Maximilian Nickel, Matt Le - Article

🐍 [2] Flow Matching in 100 LOC by François Rozet - Implementation

🐍 [3] Zuko package - Website