POT : Python Optimal Transport

PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)

(ICASSP'21/CVPR'21) Wasserstein Barycenter Transport

Source code for the ICML2019 paper "Subspace Robust Wasserstein Distances"

Fast Topological Clustering with Wasserstein Distance (ICLR 2022)

Python Implementation of "Fast Computation of Wasserstein Barycenters"

(PhD Thesis) Multi-Source Domain Adaptation through Wasserstein Barycenters

Implementation of Sliced Wasserstein distance, SW barycenters and application to K-means clustering.

Code for "Fixed Support Tree-Sliced Wasserstein Barycenter"

Distributional crypto-return forecasting via Wasserstein-geodesic extrapolation in quantile-function space. WGeo family wins 12/12 (asset × horizon) cells over 6.75y walk-forward CRPS vs GARCH and classical baselines. v0.4.

Implementation of Van Dijcke, David. "Regression Discontinuity Design with Distribution-Valued Outcomes." Manuscript.

'Wasserstein Dictionaries for Persistence Diagrams' - TDA MVA'25

Repository to store code/data for implementing OT based Column-Concentrated sampling technique and matrix completion algorithms.