Optimal Transport Python Packages

pot

POT : Python Optimal Transport

437K 3K 545

point-cloud-utils

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

108K 2K 120

ott-jax

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

30K 737 134

torchcfm

TorchCFM: a Conditional Flow Matching library

17K 2K 212

torchdr

TorchDR - PyTorch Dimensionality Reduction

4K 202 14

squarenet

Point cloud to grid mapping for machine learning (CNN-ready)

2K 3 1

moscot

Multi-omic single-cell optimal transport tools

2K 207 16

wot

A software package for analyzing snapshots of developmental processes

1K 162 38

mosaicraft

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

1K 0 0

pyapd

A Python library for generating (optimal) anisotropic power diagrams

890 17 1

catsbench

A Benchmark for Categorical-State Schrödinger Bridges and Entropic Optimal Transport

877 3 0

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.

854 4 0

pilotgm

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

770 9 3

torchcdt

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

748 1 0

transmorph

A unifying data integration framework.

554 11 2

fugw

A collection of gpu-compatible solvers for fused unbalanced gromov-wasserstein optimization problems

460 49 6

ot1d

OT1D: Discrete Optimal Transport in 1D by Linear Programming

429 9 1

torchgw

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

411 2 0