Learn to Evolve: Self-supervised Neural JKO Operator for Wasserstein Gradient Flow

Xue Feng, Li Wang, Deanna Needell +1

The Jordan-Kinderlehrer-Otto (JKO) scheme provides a stable variational framework for computing Wasserstein gradient flows, but its practical use is often limited by the high compu…