ot-triton-lab / flash-sinkhornView on GitHub
FlashSinkhorn: IO-Aware Entropic Optimal Transport in PyTorch + Triton. Streaming Sinkhorn with O(nd) memory.
183Mar 3, 2026Updated last week

Alternatives and similar repositories for flash-sinkhorn

Users that are interested in flash-sinkhorn are comparing it to the libraries listed below

Sorting:

Are these results useful?