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_genai-0.7.2-cp312-cp312-manylinux1_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | fef6c94755bff9de49f8d2a2d0c1e05e5c7056d03c4989a32b6cf272e539c767 |
|
MD5 | 4ec1c5c57f26249b0e7d569323749d90 |
|
BLAKE2b-256 | f6a09567077b7c800e19b026bb1abcb0bf48c0dfd79c8ba6c3d1392ea9558ec4 |
Hashes for fbgemm_gpu_genai-0.7.2-cp311-cp311-manylinux1_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 09dc84d9d95d6e9d63bc23470c6f1e2dfd83de86e631d19d04230d08a77030ae |
|
MD5 | 509ab15d32289b550a8e76cc46f75d05 |
|
BLAKE2b-256 | 4eff6260d10d4b00177579300fec8a2b26bd8391ef6fd149df23086496c06916 |
Hashes for fbgemm_gpu_genai-0.7.2-cp310-cp310-manylinux1_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 5e6dff4a79339f27241f247d77462e5fceb7982a32726bfc266ea15fed658327 |
|
MD5 | a824c02472a505903b1488a35a65865a |
|
BLAKE2b-256 | 543eed66e8d2c66524cb171a92d3a2637789ead21f51a5826412adbf0c899fd8 |