SparseOperationKit (SOK) is a python package wrapped GPU accelerated operations dedicated for sparse training / inference cases.
Project description
SparseOperationKit (SOK) is a python package wrapped GPU accelerated operations dedicated for sparse training / inference cases. It is designed to be compatible with common DeepLearning (DL) frameworks, for instance, TensorFlow. Most of the algorithm implementations in SOK are extracted from HugeCTR, which is a GPU-accelerated recommender framework designed to distribute training across multiple GPUs and nodes and estimate Click-Through Rates (CTRs). If you are looking for a very efficient solution for CTRs, please check HugeCTR.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Hashes for sparse_operation_kit-linux-any-1.1.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | c8fdab4bf7a6257b63cdb6cd635df53359b97456a1230c732c5f01d18a16519b |
|
MD5 | bd9565057445e6d9280afe720510c03d |
|
BLAKE2b-256 | 5a19ec1c56b136acfac2f93e471ec8c3a0ca2bd0ed5c00dbcb42ce30f4389feb |