Code

Project repositories

1. Code for “Plug-in estimation of Schrödinger bridges” (Pooladian and Niles-Weed, 2024) Python
2. Code for “Algorithms for mean-field variational inference via polyhedral optimization in the Wasserstein space” (Jiang, Chewi, and Pooladian, 2023) Python
3. Code for “Entropic estimation of optimal transport maps” (Pooladian and Niles-Weed, 2021) can be obtained by merging the tools found in this repo and this repo.
4. Code for “Debiaser Beware: Pitfalls in Centering Regularized Transport Maps” (Pooladian, Cuturi, and Niles-Weed, ICML 2022) Python

Implementations of other things

1. Implementation of a 1-Nearest Neighbor estimator for optimal transport maps as seen in Plugin Estimation of Smooth Transport Maps (Manole et al., 2021) Python