Thanks all who helped out and commented in this thread with tips and tricks!

I gave a talked recently on some work we have been doing in the lab that incorporated aspects from several of the responses.

docs.google.com/presentation...

#DiffusionModels #FlowMatching #StochasticInterpolants