No project description provided
Project description
FBGEMM_GPU
FBGEMM_GPU (FBGEMM GPU Kernels Library) is a collection of high-performance PyTorch GPU operator libraries for training and inference. The library provides efficient table batched embedding bag, data layout transformation, and quantization supports.
FBGEMM_GPU is currently tested with CUDA 12.1 and 11.8 in CI, and with PyTorch packages (2.1+) that are built against those CUDA versions.
See the full Documentation for more information on building, installing, and developing with FBGEMM_GPU, as well as the most up-to-date support matrix for this library.
Join the FBGEMM_GPU Community
For questions, support, news updates, or feature requests, please feel free to:
- File a ticket in GitHub Issues
- Post a discussion in GitHub Discussions
- Reach out to us on the
#fbgemm
channel in PyTorch Slack
For contributions, please see the CONTRIBUTING
file for
ways to help out.
License
FBGEMM_GPU is BSD licensed, as found in the LICENSE
file.
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 Distributions
Built Distributions
Hashes for fbgemm_gpu-0.8.0-cp312-cp312-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | de03004a83950667175ea45b5de8e5ee9bdd33d38ea3ce9f9b1f98db8559a771 |
|
MD5 | 2bcf301702535df3c46c016bbc12d952 |
|
BLAKE2b-256 | 70beee542c71a458512e87b769117da7a9990bf2a04f63ee204cf564e9241d3f |
Hashes for fbgemm_gpu-0.8.0-cp311-cp311-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | aba861138e0e1f2afb879121ecd0a4014a4e31314ad4cfe7ffc5356d818942d5 |
|
MD5 | 53222b3ea88cff8e080e2f3cf15c6a4d |
|
BLAKE2b-256 | 69f366a8e257b45effa049ecb4082214a5c448703d9591e47f3f47dda9c5f38e |
Hashes for fbgemm_gpu-0.8.0-cp310-cp310-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 9956e7d7462e8fad087c5ac9801a249c564ab5c6a7be1c23159517da2e960a5b |
|
MD5 | c667bc30a059e6d124e42dc26fb0bb25 |
|
BLAKE2b-256 | 8742dec9eccbcf4eb720f5b889becf1a171f5464447f9251d26e933c3e41e143 |
Hashes for fbgemm_gpu-0.8.0-cp39-cp39-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 65b48833c6c72d05c95f472e7a0aba26c323b3e6687ca8fa2df667316814690e |
|
MD5 | 2e0e4b38f8ee7f7b4c760feb4ef0f5eb |
|
BLAKE2b-256 | 3ad1cbc5723eb989fa8183b28ff59bc3a83b0686f842c6b29af0fd51f5f13324 |
Hashes for fbgemm_gpu-0.8.0-cp38-cp38-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | cd1f96a902c4d1adc14ca1d0f80ea80418a0eb52cc1afcb2576f7d56538b7558 |
|
MD5 | 180a2e37980b5465238a1764b1a8be82 |
|
BLAKE2b-256 | 3bd0ad37ba90f6e311b7a24bdfd742ffc6dde6f46bd15f1aeab21ecdf9c580f9 |