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

437K 3K 545
ott-jax
ott-jax

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

30K 737 134
chansigit
torchgw

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

411 2 0