Optimal Transport ML ML Frameworks
PyTorch implementations of optimal transport theory and algorithms, including neural optimal transport, Wasserstein distances, barycenters, and gradient flows. Does NOT include general deep learning frameworks, transportation logistics, or non-optimal-transport mathematical optimization methods.
There are 14 optimal transport ml frameworks tracked. 1 score above 70 (verified tier). The highest-rated is PythonOT/POT at 81/100 with 2,772 stars. 1 of the top 10 are actively maintained.
Get all 14 projects as JSON
curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=optimal-transport-ml&limit=20"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
| # | Framework | Score | Tier |
|---|---|---|---|
| 1 |
PythonOT/POT
POT : Python Optimal Transport |
|
Verified |
| 2 |
recursionpharma/gflownet
GFlowNet library specialized for graph & molecular data |
|
Emerging |
| 3 |
IShengFang/SpectralNormalizationKeras
Spectral Normalization for Keras Dense and Convolution Layers |
|
Emerging |
| 4 |
dfdazac/wassdistance
Approximating Wasserstein distances with PyTorch |
|
Emerging |
| 5 |
iamalexkorotin/NeuralOptimalTransport
PyTorch implementation of "Neural Optimal Transport" (ICLR 2023 Spotlight) |
|
Emerging |
| 6 |
PetrMokrov/Large-Scale-Wasserstein-Gradient-Flows
Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021) |
|
Emerging |
| 7 |
joisino/otbook
書籍『最適輸送の理論とアルゴリズム』のサポートページです。 |
|
Emerging |
| 8 |
rhfeiyang/PPOT
Official implementation of 'P$^2$OT: Progressive Partial Optimal Transport... |
|
Experimental |
| 9 |
iamalexkorotin/Wasserstein2Barycenters
PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter... |
|
Experimental |
| 10 |
iamalexkorotin/KernelNeuralOptimalTransport
PyTorch implementation of "Kernel Neural Optimal Transport" (ICLR 2023) |
|
Experimental |
| 11 |
milenagazdieva/OT-Super-Resolution
Pytorch implementation of "An Optimal Transport Perspective on Unpaired... |
|
Experimental |
| 12 |
milenagazdieva/U-NOTBarycenters
PyTorch implementation of "Robust Barycenter Estimation using... |
|
Experimental |
| 13 |
iamalexkorotin/WassersteinIterativeNetworks
PyTorch implementation of "Wasserstein Iterative Networks for Barycenter... |
|
Experimental |
| 14 |
rhfeiyang/SPPOT
Official implementation of 'SP$^2$OT: Semantic-Regularized Progressive... |
|
Experimental |