flashlib

FlashLib 是一个面向经典机器学习算子的高性能 GPU 加速库，提供 KMeans、KNN、PCA、SVD、DBSCAN、UMAP 等 15+ 算子的内存高效实现。基于 Triton 和 CuteDSL 构建，支持多精度 GEMM 及低开销运行时预测（无需 GPU 即可估算 FLOPs/HBM），所有接口兼容 PyTorch 张量并提供 sklearn 风格类封装。适用于大规模向量聚类、降维、相似性检索及特征工程等需高频调用基础 ML 算子的场景，尤其适合在 GPU 资源受限或需精细性能预算的生产 pipeline 中部署。