mlxVS:Apple Silicon 上使用 MLX 的 cuVS 式 GPU 最近邻搜索(NN-Descent、CAGRA)[P]
原文:mlxVS: cuVS-style GPU nearest-neighbor search (NN-Descent, CAGRA) on Apple Silicon with MLX [P]
我发布了 mlxVS,这是一个开源 (Apache-2.0) 库,它在 Apple Silicon 上运行来自 NVIDIA cuVS 的主要 GPU 最近邻算法,构建于 MLX 之上。它涵盖精确的 k-NN、k-means、NN-Descent k-NN 图和 CAGRA(图修剪加 GPU 图搜索),以及来自 Python(NumPy in)或 C++ 的 cuVS 参数名称。结果(SIFT1M、985k x 128、L2、MacBook Pro M3 Pro:18 核 GPU、12 核 CPU)
正文没能搬到本站:该站禁止服务器(机房 IP)抓取,返回 site,换任何浏览器标识都拿不到。在手机上打开原站往往又慢又卡,下面先把收录到的内容给你。
本站会持续尝试重新抓取;也可以收藏这条,正文抓到后会永久保留。
正文由本站服务器抓取保存(已自动过滤广告、推荐位与追踪图片),排版与原站可能不同;版权归原站所有。