Sinkhorn optimal transport kernels in PyTorch + Triton (squared Euclidean, no cost matrix materialization).
[ICLR 2026] A Benchmark for Categorical-State Schrödinger Bridges and Entropic Optimal Transport
