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