Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)
deep-learning
pytorch
optimal-transport
nonlinear-filtering
fokker-planck-equation
bayesian-logistic-regression
wasserstein-gradient-flows
-
Updated
May 8, 2022 - Jupyter Notebook