We modify the JKO scheme, which is a time discretization of the Wasserstein gradient flow, by replacing the Wasserstein distance with more general transport costs on manifolds. We show when the cost function has a mixed Hessian which defines a Riemannian metri... ...
