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_nightly-2024.11.12-cp312-cp312-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | f73872c451ea66a1da206db478ae8b9e3c48199c0589fd51b150196d3d23a709 |
|
MD5 | 4aef7f1cf3e55aa30af3422ddbacced2 |
|
BLAKE2b-256 | a67103bfbcd48d44bea779d9f6a19ac379b008bd0566dc8dd79476d110a16d48 |
Hashes for fbgemm_gpu_nightly-2024.11.12-cp311-cp311-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 11923771f8de9fb02a2096f839cefdf78320e9bfa34bc3d46741497192e97dfe |
|
MD5 | 0581fac4c85dfea1910b220361d06067 |
|
BLAKE2b-256 | 2d91ad98bf4657ea58bc6eae4bc9413a14ee10d6d73352579ba8348edcb2913e |
Hashes for fbgemm_gpu_nightly-2024.11.12-cp310-cp310-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0cbdd462ae5947286cb98589ab6f70ea5f9371bcc5722c2805aeb4ab3b709911 |
|
MD5 | 5d50bfc525b89cbb86f6d6ffa106946d |
|
BLAKE2b-256 | 18e1888401685ea14a49c0d2f138d245b5c3b7d40d68ed5885efb0a0d6831a5e |
Hashes for fbgemm_gpu_nightly-2024.11.12-cp39-cp39-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 60a0202ba388646523436c56d95466fff053c8acdffbd963cd6e758590d2b2eb |
|
MD5 | 56a882b40866622610c32c9b55ae9acd |
|
BLAKE2b-256 | 324c5507c08199cf7f016e67670215f0d5575cba5f579f1d9e0f17cc18c73c77 |