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
File details
Details for the file merlin-sok-2.1.0.tar.gz
.
File metadata
- Download URL: merlin-sok-2.1.0.tar.gz
- Upload date:
- Size: 8.7 MB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.0 CPython/3.10.12
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | bd744499effd3bd4e17c95a7bc7eb30cdb89b92eec10313e8f428565a6a02b61 |
|
MD5 | f939583b7eb876bb080f2e53000b9982 |
|
BLAKE2b-256 | 4f4453127de1bdd503adbede6350f827612b17f32e5551b7737227ab2f433d0b |