gromov-wasserstein
POT : Python Optimal Transport
Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.
TorchGW — Fast Sampled Gromov-Wasserstein optimal transport in pure PyTorch. GPU-accelerated with Triton fused Sinkhorn kernels. 3-175x faster than POT.