Repository containing code for "Debiaser Beware: Pitfalls of Centering Regularized Transport Maps" (ICML 2022); this is joint work with Jonathan Niles-Weed, and Marco Cuturi.
Code should work out of the box in Google Colab. We recommend enabling GPU support, otherwise the runtime becomes quite slow for large sample sizes.
Our code is based off the following repository for the paper "Faster Wasserstein Distance Estimation with the Sinkhorn Divergence" by Chizat, Roussillon, Léger, Vialard, and Peyré (2020).