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_cpu-2024.10.10-cp312-cp312-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 999bee76b140a01342178aad21c5f44c14caa88c8c4a8e579c6ed55f0c52a187 |
|
MD5 | 575ae5b9622ac3071d3fa140a3e4a445 |
|
BLAKE2b-256 | 94c61f6da7a2b81225fed31128af18c395310330ab635a3ea7371bfc30191069 |
Hashes for fbgemm_gpu_nightly_cpu-2024.10.10-cp312-cp312-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 864bbf3a0db697f17d2ec44d328a8d054048e3eacc1852128dcb7158e90d0f81 |
|
MD5 | f5c6785a64bb6703da4e43dc0dc38cb5 |
|
BLAKE2b-256 | 7421f026c2b029e355d270ddf64bb2ef71c31756d2420876df8ed1a8d43df4f9 |
Hashes for fbgemm_gpu_nightly_cpu-2024.10.10-cp311-cp311-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 6ae733a9236e8f58c8e31412d5122b25e17a4a8a3b1e7cfcb2fb52a58f80e4c0 |
|
MD5 | 792409903f2c081ddaf3aff7943ca3cc |
|
BLAKE2b-256 | 237058e6a720a521cbadfb724d1d404745e63242d5a65018f1ee7eed84cdfccb |
Hashes for fbgemm_gpu_nightly_cpu-2024.10.10-cp311-cp311-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e53a3bb166269487d01aa5cb655a0487faa075125000891b88bdb4fe6ef8fc65 |
|
MD5 | 6fc5ec34d0176820afb37feb32b2542b |
|
BLAKE2b-256 | 45d775cc81ea55dc27361dd8c2a2f97fcfcf70ca02c1babf468bbe0e7bd36fc5 |
Hashes for fbgemm_gpu_nightly_cpu-2024.10.10-cp310-cp310-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0c503b42838bcf4febc7ebeee9ed732f449f6f0ed5c5d1601fbf1f91d8e62f6f |
|
MD5 | e96fe3974384201e82d89986ac042a64 |
|
BLAKE2b-256 | 96e52716e912e4fcf78cca6b86df394f834d1608afd361c584edf630de39032d |
Hashes for fbgemm_gpu_nightly_cpu-2024.10.10-cp310-cp310-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1cd76b2018c16090c50118a3d2c35aec0846794c315e60586841cad4ab3522b4 |
|
MD5 | e3d722b948dbaf82f75e59e09de09bd3 |
|
BLAKE2b-256 | f31eb063940a6f44b9aae224267c4e38451dde71ab27a4d1a745971ec6e71af0 |
Hashes for fbgemm_gpu_nightly_cpu-2024.10.10-cp39-cp39-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | ae8962e81cdfcb7b7482b9d420e4ad6aad4cdd0c2431a9e1493fb8981df9949f |
|
MD5 | 7c42f4c27c029346ee6cc85854876291 |
|
BLAKE2b-256 | c2d17ad9338cdae8ff8082f6a67f4c9cc4b21013eca23ed9fe2fbf1f1b14ada2 |
Hashes for fbgemm_gpu_nightly_cpu-2024.10.10-cp39-cp39-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1e201a8685a6976ff44b6c2487835c651c9afde4b613ea806bd2eb66138c24aa |
|
MD5 | 970373b2b4052acae5508a61d6497450 |
|
BLAKE2b-256 | dd5fcd1216d0be61f1225f9317d4eab9529d3bddd262d49b4744f327afeea8e5 |