FlashSinkhorn 2: Block-Sparse Entropic Optimal Transport
FS2 is a solver for squared-Euclidean cost on low-dimensional point clouds that solves large discrete EOT problems to a prescribed marginal residual on a single GPU by coupling two stages.
Felix X.-F. Ye, Yu-Chin-Fa-Bian Lim, Nai-Gang Wang et al.
· 0 citations