Deep JKO: time-implicit particle methods for general nonlinear gradient flows

Wonjun Lee, Li Wang, Wuchen Li

We develop novel neural network-based implicit particle methods to compute high-dimensional Wasserstein-type gradient flows with linear and nonlinear mobility functions. The main i…