The official repository of FlashSinkhorn [ICML 2026 Spotlight]
machine-learning gpu cuda pytorch triton optimal-transport sinkhorn flash-attention entropic-optimal-transport flashsinkhorn
-
Updated
May 20, 2026 - Python