Large-Scale Wasserstein Gradient Flows - Mokrov, Petr; Korotin, Alexander; Li, Lingxiao; Genevay, Aude; Solomon, Justin; Burnaev, Evgeny (Advances in Neural Information Processing Systems, 2021)