Sinkhorn Python Packages

point-cloud-utils

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

ott-jax

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

mhc-mlx

MLX + Metal implementation of mHC: Manifold-Constrained Hyper-Connections by DeepSeek-AI.

torchgw

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

flash-sinkhorn

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