PyRank

Gromov Wasserstein Python Packages

Python packages with the GitHub topic gromov-wasserstein. Sorted by monthly downloads, with stars and forks.
PythonOT
pot

POT : Python Optimal Transport

331K 3K 571
ott-jax
ott-jax

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

35K 764 139
chansigit
torchgw

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

72 6 1