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.

flash-sinkhorn

The official repository of FlashSinkhorn [ICML 2026 Oral]

torchgw

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