跳转到主要内容

SparseOperationKit (SOK) 是一个Python包,用于包装针对稀疏训练/推理案例的GPU加速操作。

项目描述

SparseOperationKit (SOK) 是一个Python包,用于包装针对稀疏训练/推理案例的GPU加速操作。它设计为与常见的深度学习(DL)框架兼容,例如TensorFlow。SOK中的大多数算法实现都是从HugeCTR提取的,HugeCTR是一个用于在多个GPU和节点上分布训练并估计点击通过率(CTR)的GPU加速推荐框架。如果您正在寻找一个非常高效的CTR解决方案,请检查HugeCTR。

项目详情


下载文件

下载适用于您平台文件的文件。如果您不确定选择哪个,请了解更多关于安装包的信息。

源分发

merlin-sok-2.1.0.tar.gz (8.7 MB 查看散列值)

上传时间:

支持