PyRank

Optimal Transport Python Packages

Python packages with the GitHub topic optimal-transport. Sorted by relevance, with stars and monthly downloads.
PythonOT
pot

POT : Python Optimal Transport

517K 3K 547
fwilliams
point-cloud-utils

An easy-to-use Python library for processing and manipulating 3D point clouds and meshes.

112K 2K 120
ott-jax
ott-jax

Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.

71K 749 135
atong01
torchcfm

TorchCFM: a Conditional Flow Matching library

12K 3K 218
theislab
moscot

Multi-omic single-cell optimal transport tools

3K 214 16
TorchDR
torchdr

TorchDR - PyTorch Dimensionality Reduction

2K 205 14
mbuze
pyapd

A Python library for generating (optimal) anisotropic power diagrams

1K 17 2
alexisthual
fugw

Scalable python GPU solvers for fused unbalanced gromov-wasserstein optimal transport problems, with routines and examples to align brain data (fMRI)

1K 52 6
broadinstitute
wot

A software package for analyzing snapshots of developmental processes

769 164 38
hinanohart
mosaicraft

Perceptual photomosaic generator — Oklab color matching, MKL optimal transport, Hungarian placement, and Oklch tile-pool expansion

620 0 0
DaminK
ggml-ot

Improve Optimal Transport applications by Global Ground Metric Learning (published at AISTATS2025). This package supports single-cell data in Anndata format and generic array-like data.

557 4 0
Risitop
transmorph

Computational framework for dataset integration

498 12 2
gregkseno
catsbench

[ICLR 2026] A Benchmark for Categorical-State Schrödinger Bridges and Entropic Optimal Transport

454 10 0
ot-triton-lab
flash-sinkhorn

The official repository of FlashSinkhorn [ICML 2026 Oral]

393 203 20
chansigit
torchgw

TorchGW — Fast Sampled Gromov-Wasserstein optimal transport in pure PyTorch. GPU-accelerated with Triton fused Sinkhorn kernels. 3-175x faster than POT.

378 5 0
tomluetjen
torchcdt

A differentiable implementation of the (Radon-) Cumulative Distribution Transform in PyTorch

372 1 0
AI4SCR
cmonge

Conditional Monge Gap (2026; Nature Machine Intelligence)

369 16 1
stegua
ot1d

OT1D: Discrete Optimal Transport in 1D by Linear Programming

365 9 1
LvDAO
jgot

Dynamic optimal transport on sparse reversible graphs in JAX

361 4 0
CostaLab
pilotgm

Patient-Level Analysis of Single Cell Disease Atlas with Optimal Transport of Gaussian Mixtures Variational Autoencoders

321 10 3
yzf072
spatrack

An optimal transport-based algorithm for inferring cell trajectories of spatial transcriptomics.

225 42 4
gjhuizing
wsingular

Python package for the ICML 2022 paper "Unsupervised Ground Metric Learning Using Wasserstein Singular Vectors".

184 10 1
kuslavicek
ballot

High-performance Python package for balanced k-means clustering using optimal transport and entropic regularization

180 2 1
sacha-ichbiah
optiflow

Reconstruction of flows from kymographs with optimal transport

166 4 0