Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)
-
Updated
May 8, 2022 - Jupyter Notebook
Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)
[ICLR 2026] JAX implementation of the iJKOnet method
Discretized Wasserstein Particle Flows of a MMD-regularized f-divergence functional.
Reconstructs continuous gene expression dynamics from static scRNA-seq snapshots using Wasserstein gradient flows and a particle-based explicit method. Applied to EMT, stem cell differentiation, and breast cancer treatment response.
Add a description, image, and links to the wasserstein-gradient-flows topic page so that developers can more easily learn about it.
To associate your repository with the wasserstein-gradient-flows topic, visit your repo's landing page and select "manage topics."