注意

RAFT 中的向量搜索和聚类算法正在迁移到一个专门用于向量搜索的新库 cuVS。在此迁移过程中,我们将继续支持 RAFT 中的向量搜索算法,但在 RAPIDS 24.06(六月)版本发布后将不再更新它们。我们计划在 RAPIDS 24.10(十月)版本发布前完成迁移,并在 24.12(十二月)版本中将它们从 RAFT 中完全移除。

RAPIDS RAFT: 用于向量搜索及其他功能的通用加速函数和工具#

RAFT Tech Stack

有用资源#

什么是 RAFT?#

RAFT 包含用于机器学习和信息检索的基础性常用算法和原语。这些算法经过 CUDA 加速,是更轻松地编写高性能应用程序的构建模块。

通过采用基于原语的算法开发方法,RAFT 可以

  • 加速算法构建时间

  • 通过最大限度地跨项目重用,减轻维护负担,并且

  • 集中核心可重用计算,使未来的优化能惠及所有使用这些计算的算法。

虽然不尽全面,但以下通用类别有助于总结 RAFT 包含的加速构建模块

类别

示例

数据格式

稀疏与稠密格式,转换,数据生成

稠密运算

线性代数,矩阵和向量运算,切片,范数,因式分解,最小二乘法,SVD 和特征值问题

稀疏运算

线性代数,特征值问题,切片,范数,归约,因式分解,对称化,组件与标记

求解器

组合优化,迭代求解器

统计

抽样,矩和汇总统计,指标

工具与实用程序

用于开发 CUDA 应用程序的常用实用程序,多节点多 GPU 基础设施

索引和表格#