Langevin Monte Carlo and JKO splittingBernton · Proceedings of the 31st Conference On Learning Theory, PMLR 75:1777-1798, 20182018PDF
A JKO Splitting Scheme for Kantorovich--Fisher--Rao Gradient FlowsGallouët · Society for Industrial and Applied Mathematics2017PDF369 KB
Taming Hyperparameter Tuning in Continuous Normalizing Flows Using the JKO SchemeVidal · arXiv2022PDF
Forward-backward Gaussian Variational Inference via JKO in the Bures-Wasserstein SpaceDiao · arXiv2023PDF