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-0.7.0-cp312-cp312-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1580b94fbc20471bb47b428ca3cfb02c6de8d3c607ab5b72737dae8e72f116c3 |
|
MD5 | 4f504c1a773d47aa869dae252b3a3af7 |
|
BLAKE2b-256 | e2ffcde7940fb90d2019546e87f9e88efd225521d8a704ec0e172b0ae76f0b30 |
Hashes for fbgemm_gpu-0.7.0-cp311-cp311-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 638b02f46cddbfee936799ab0e18ae21cfacf90c7a132b75150e9756be47ede0 |
|
MD5 | e9d78d98b27b313d27e2e8067c321edd |
|
BLAKE2b-256 | 9f02a6db19b97e22f21d043514f73dd178da88d3b2d25c438623ccbe7f8e0f8a |
Hashes for fbgemm_gpu-0.7.0-cp310-cp310-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4077497f455a73ecdfc99f180088bb94eb1efca5d20062ecc5b5fdf0b4c48e2e |
|
MD5 | 32ad2184678a5203ae23002572a775b8 |
|
BLAKE2b-256 | 50e9a5d008e5dc8ae610a248ea78523f35d27fd012742cbd44e79f63ab194c88 |
Hashes for fbgemm_gpu-0.7.0-cp39-cp39-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 51c40ee96c30d7c4cc4eabb091a8c628b31822469bb841a6bdff8b0d6beea64c |
|
MD5 | 4ce6b72aa1e917a75c086dc1f9b67ecd |
|
BLAKE2b-256 | 7bb28f15982b6e7487eec819564fc033b234c928a5e6653177eb75da1f996a60 |
Hashes for fbgemm_gpu-0.7.0-cp38-cp38-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2e0570519c3e44df6a633a51450c58b08ed4b14b1605fa83c12581b4295fb992 |
|
MD5 | e874a91629f6346255f4fe10860288ca |
|
BLAKE2b-256 | 6995cf1466c8a4d39a2e710b03602deab21147e5a73ad625ce965236f3b41f5b |