Minimal Stochastic Interpolant

Nguyen Minh Hieu

hieuristics [at] gmail.com

The entire implementation of flow-matching fits into a screenshot!

note: some codes have been inspired by the original code base. This repo aims to simplify the core part of each algorithm.

Validation Result

Implemented the following interpolant:

• Flow Matching [Lipman et al., 2022]
• Schrodinger Bridge (I2SB) [Liu et al. 2023]

Validation Result

*training tested on M1 Mac.

Flow Matching (training time: 1.02 mins)

Schrodinger Bridge (training time: 3.45mins)