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.7.24-cp312-cp312-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 14ca03cb0b198f522f049bd1437ad3393460ba4392bd407d6dd1f57ae601fdd0 |
|
MD5 | 627dc1e5814d5bf9e4e1e6b3e96f73e6 |
|
BLAKE2b-256 | fee759afc5f42e4ebba70908263b0f4f2bbd48d3d46f4950fe3a5365f07a519a |
Hashes for fbgemm_gpu_nightly-2024.7.24-cp311-cp311-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 752850bb78ab9194033190fb796ca3c74a79e6eb8c68a5c45158d67ab7450518 |
|
MD5 | bdd728dbf4e060e949dcff9d717717ee |
|
BLAKE2b-256 | e09d839ecc822d9a269d755942ff3785655bc484a6a2ad0cf7fe1a6ee207a9ec |
Hashes for fbgemm_gpu_nightly-2024.7.24-cp310-cp310-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2cf715776f21805b16e67dbe984f04d48024cd751748f201a4031e8232b2047a |
|
MD5 | c1720b566940779ce693b797864a8c31 |
|
BLAKE2b-256 | dd6e44a9c29d7d3af93598ebc66020c9101df63b34fcf8ee18986410e3d21aef |
Hashes for fbgemm_gpu_nightly-2024.7.24-cp39-cp39-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4099d192b28fe468726d42999f1d3918e4538b19068ebc5b39ce12b4eab2dc77 |
|
MD5 | 9e8813c29e6698835b29062216ff4885 |
|
BLAKE2b-256 | 9de15de06927f2674e32b78c750cbb83c9524819813fc26387437ba290771876 |
Hashes for fbgemm_gpu_nightly-2024.7.24-cp38-cp38-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d22293e7ef8612e51e159115152c4fbf8f8ebb350f736f9b5ab9f1b00dcd225b |
|
MD5 | 67306fd364e434778d8ab7b2785c5824 |
|
BLAKE2b-256 | 6bc55c447a7c60d4568022ccf1668abcd91434684d204bf143eb982a9b575d03 |