This Jupyter notebook generates the Figures of the paper "The Wasserstein Proximal Gradient Algorithm"

Just launch the cells sequentially. You can also play with the parameters e.g. niter (number of iterations), d (number of particles), dim (dimension of the space) etc.