Some publicly available code for my projects. Though, for anything related to optimal transport (with entropic regularization), I recommend the Optimal Transport Toolbox (OTT).

Some poorly written code

  1. Code for “Debiaser Beware: Pitfalls in Centering Regularized Transport Maps” (Pooladian, Cuturi, and Niles-Weed, ICML 2022) [Python]
  2. Code for “Entropic estimation of optimal transport maps” (Pooladian and Niles-Weed, 2021) is found by combining this repo and this repo
  3. Implementation of 1-Nearest Neighbor estimator for optimal transport maps as seen in Plugin Estimation of Smooth Transport Maps (Manole et al., 2021) [Python]