PyPI Stats
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About
Home

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

424K 3K 544
fwilliams
point-cloud-utils

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

152K 2K 120
ott-jax
ott-jax

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

28K 735 134
atong01
torchcfm

TorchCFM: a Conditional Flow Matching library

17K 2K 210
TorchDR
torchdr

TorchDR - PyTorch Dimensionality Reduction

4K 202 14
ArmanddeCacqueray
squarenet

Sparse local operations for point clouds in any dimension.

3K 1 0
theislab
moscot

Multi-omic single-cell optimal transport tools

2K 206 16
hinanohart
mosaicraft

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

2K 0 0
broadinstitute
wot

A software package for analyzing snapshots of developmental processes

1K 162 38
chansigit
torchgw

Fast Sampled Gromov-Wasserstein optimal transport solver — pure PyTorch, scalable, differentiable

1K 1 0
mbuze
pyapd

Computing (optimal) anisotropic power diagrams using GPU acceleration

813 17 1
DaminK
ggml-ot

Supervised OT with Global Ground Metric Learning

743 4 0
Risitop
transmorph

A unifying data integration framework.

642 11 2
tomluetjen
torchcdt

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

604 1 0
alexisthual
fugw

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

394 49 6
stegua
ot1d

OT1D: Discrete Optimal Transport in 1D by Linear Programming

390 9 1
ot-triton-lab
flash-sinkhorn

Sinkhorn optimal transport kernels in PyTorch + Triton (squared Euclidean, no cost matrix materialization).

365 189 19
LvDAO
jgot

Dynamic optimal transport on sparse reversible graphs in JAX

304 4 0
AI4SCR
cmonge

Learning conditional optimal transport maps

275 11 1
BorgwardtLab
wtk

A Wasserstein Subsequence Kernel for Time Series.

265 21 7
gjhuizing
wsingular

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

259 10 1
yzf072
spatrack

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

219 39 4
sacha-ichbiah
optiflow

Reconstruction of flows from kymographs with optimal transport

133 4 0
burning-cost
insurance-fairness-ot

Discrimination-free insurance pricing via Lindholm marginalisation, causal path decomposition, and Wasserstein barycenter correction

119 0 0
    • Data from PyPI, GitHub, ClickHouse, and BigQuery