/Optimal-Flow-Matching

The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)

Primary LanguageJupyter Notebook

Issues