ot-triton-lab/flash-sinkhorn
The official repository of FlashSinkhorn [ICML 2026 Oral]
GitHub repository with 198 stars and 20 forks.
Language: Python
Topics: cuda, entropic-optimal-transport, flash-attention, gpu, machine-learning, optimal-transport, pytorch, sinkhorn, triton, flashsinkhorn