ot-triton-lab/flash-sinkhorn
FlashSinkhorn: IO-Aware Entropic Optimal Transport in PyTorch + Triton. Streaming Sinkhorn with O(nd) memory.
When comparing two sets of data points, like images or molecular structures, FlashSinkhorn quickly calculates the 'distance' or similarity between them. It takes in your data points and outputs a cost or a plan for how to transform one set into the other, even for very large datasets. This is designed for scientists, engineers, or researchers working with high-dimensional data.
183 stars. Available on PyPI.
Use this if you need to compare large collections of data points efficiently, such as for machine learning tasks, image analysis, or molecular shape comparison, and memory usage is a concern.
Not ideal if your data is small enough that standard methods work fine, or if you don't have access to a CUDA-enabled GPU.
Stars
183
Forks
19
Language
Python
License
MIT
Category
Last pushed
Mar 03, 2026
Commits (30d)
0
Dependencies
3
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/ot-triton-lab/flash-sinkhorn"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related frameworks
triton-inference-server/server
The Triton Inference Server provides an optimized cloud and edge inferencing solution.
gpu-mode/Triton-Puzzles
Puzzles for learning Triton
hailo-ai/hailo_model_zoo
The Hailo Model Zoo includes pre-trained models and a full building and evaluation environment
open-mmlab/mmdeploy
OpenMMLab Model Deployment Framework
hyperai/tvm-cn
TVM Documentation in Chinese Simplified / TVM 中文文档