JKO schemes with general transport costs

Cale Rankin, Ting-Kam Leonard Wong

We modify the JKO scheme, which is a time discretization of Wasserstein gradient flows, by replacing the Wasserstein distance with more general transport costs on manifolds. We sho…