#

flashsinkhorn

Here is 1 public repository matching this topic...

ot-triton-lab / flash-sinkhorn

FlashSinkhorn: IO-Aware Entropic Optimal Transport in PyTorch + Triton. Streaming Sinkhorn with O(nd) memory.

machine-learning gpu cuda pytorch triton optimal-transport sinkhorn flash-attention entropic-optimal-transport flashsinkhorn

Updated Mar 3, 2026
Python

Improve this page

Add a description, image, and links to the flashsinkhorn topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the flashsinkhorn topic, visit your repo's landing page and select "manage topics."