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