@neural_avb: 深度学习兄弟姐妹们，别错过这个。你可以在嵌入空间中对数百万文档进行聚类，批量注释…

X AI KOLs Timeline 工具
gpu-library classical-ml clustering kmeans pca tsne open-source

摘要

Shuo Yang 及其团队发布了 FlashLib，这是一个 GPU 库，可以加速 KMeans、KNN、HDBSCAN、PCA 和 t-SNE 等经典机器学习算子，声称加速比高达 208 倍。

深度学习兄弟姐妹们，别错过这个。 你可以在嵌入空间中对数百万文档进行聚类、批量注释、可视化……基本上免费且只需几秒钟。 https://t.co/PRaogzkY8J
查看原文
查看缓存全文

缓存时间: 2026/05/27 09:21

Deep learning 兄弟姐妹们，千万别错过这个。

你可以在嵌入空间中对数百万文档进行聚类、批量标注、可视化……基本上免费且只需几秒钟。https://t.co/PRaogzkY8J

Shuo Yang（@Andy_ShuoYang）： Flash-KMeans 只是个开始。

今天，来自 Flash-KMeans 团队，我们发布了 FlashLib —— 一个用于快速、可预测、可集成代理的经典机器学习算子的 GPU 库。

在 KMeans 上高达 26 倍，KNN 上 19 倍，HDBSCAN 上 40 倍，TruncatedSVD 上 208 倍，PCA 上 47 倍，精确 t-SNE 上 147 倍，

相似文章

Flash-GMM：一种用于可扩展软聚类的内存高效内核

Hugging Face Daily Papers

Flash-GMM 引入了一个用于高斯混合模型的融合Triton内核，实现了20倍加速，并能在单个GPU上训练比之前大100倍的数据集，使软聚类成为近似最近邻搜索中k-means的可行替代方案。