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.

flash-sinkhorn

The official repository of FlashSinkhorn [ICML 2026 Oral]

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.

otloss

Optimal Transport loss functions for PyTorch — Wasserstein, Sliced Wasserstein, and WGAN-GP as drop-in nn.Module replacements