PyRank

Gromov Wasserstein Python Packages

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

POT : Python Optimal Transport

517K 3K 547
ott-jax
ott-jax

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

71K 749 135
chansigit
torchgw

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

378 5 0