Skolkovo Institute of Science and Technology, Moscow Institute of Physics and Technology, Moscow, Russia; Skolkovo Institute of Science and Technology, Artificial Intelligence Research Institute, Moscow, Russia; Massachusetts Institute of Technology, Cambridge, Massachusetts, USA
Petr Mokrov, Alexander Korotin, Lingxiao Li, Aude Genevay, Justin Solomon, Evgeny Burnaev
wasserstein-2 gradient flows;JKO stepping;optimal transport;input-convex neural networks;diffusion processes
Large-Scale Wasserstein Gradient Flows