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-1.0.0-cp312-cp312-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 8b1e3a0dc7f4d53fdc06efd8f9aa34e18c1e2a7d6e6f4211d37252b55a4e8903 |
|
MD5 | 5e0a6223c72d14162fb02b1b62b96b32 |
|
BLAKE2b-256 | f29e80139b1805c81514030f2bcf4ebf8c6e8b4882260d50522bf6c09df7b7bc |
Hashes for fbgemm_gpu_genai-1.0.0-cp311-cp311-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 34fdb9e93712fd30a19372e91a852d81b2b346c37388c0344bb25a7e52c7e0ab |
|
MD5 | c238da7a3e966194ea07f18d254b7f61 |
|
BLAKE2b-256 | d3ac05ebc4f7a0b823f77a039b88b712f23db74678c7c1b1b4899924379ebe1c |
Hashes for fbgemm_gpu_genai-1.0.0-cp310-cp310-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 02a08a4827c468d62a30595036045c7c61b0d3e282f5546b5fc56e03469e4d0f |
|
MD5 | feb2cfbbe7069b485399f7aab08d8081 |
|
BLAKE2b-256 | 56943501aef668331049a6a374cae4e7d35134b58b744072ac095679aef9be94 |
Hashes for fbgemm_gpu_genai-1.0.0-cp39-cp39-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 07e012a26350460d8a0fa3cc88e8c86bfe06790c4fbf0e090aa7be3bb1dcfdde |
|
MD5 | 704d35ac0c9ba91737b49cb7c0a0ffec |
|
BLAKE2b-256 | 1dcb59cd1fbc09e2158fec58e958472ee3e7c2199bffd13092c14bea7164f4f9 |