svg-project/flash-kmeans

Flash-KMeans是基于Triton GPU内核实现的快速、内存高效的精确K-Means聚类算法。专为处理大规模数据设计，支持批量计算，性能显著优于传统实现。适用于需要高效向量聚类的深度学习场景，如Sparse VideoGen2等任务。