TorchGW — Fast Sampled Gromov-Wasserstein optimal transport in pure PyTorch. GPU-accelerated with Triton fused Sinkhorn kernels. 3-175x faster than POT.
bioinformatics graph pytorch triton single-cell st optimal-transport graph-matching gromov-wasserstein spatial-transcriptomics sinkhorn manifold-alignment transport-plan
-
Updated
Apr 7, 2026 - Python