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