FlashSinkhorn 2: Block-Sparse Entropic Optimal Transport

Felix X. -F. Ye, Yu Chin Fabian Lim, Naigang Wang +1

Streaming GPU solvers for entropic optimal transport (EOT), such as FlashSinkhorn, avoid storing the dense kernel but still evaluate all $n\times m$ point pairs in every Sinkhorn i…