atong01 / conditional-flow-matching Python 2.6K 229 TorchCFM: a Conditional Flow Matching library continuous-normalizing-flows optimal-transport pytorch