/WassersteinIterativeNetworks

PyTorch implementation of "Wasserstein Iterative Networks for Barycenter Estimation" (NeurIPS 2022)

Primary LanguageJupyter NotebookMIT LicenseMIT

Stargazers