
Implementation of the paper "Neural Wasserstein Gradient Flows for Discrepancies with Riesz Kernels". The paper is available at https://proceedings.mlr.press/v202/altekruger23a.html. The repository contains an implementation of Neural backward scheme and Neural forward scheme. It contains scripts for reproducing the numerical examples in Section 6.
