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-1.0.0-cp312-cp312-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 5189fabf06c00b6c98accf61a7ccc4b6d27f8928f2957f9f4a2cb65e4c5f84e9 |
|
MD5 | 165e5bdd9c118ba342ef4b9fc0242fc8 |
|
BLAKE2b-256 | dcf17f9a9a149a611973821f02920e9072bf0e0331842fb8a81e95ff43ef2e88 |
Hashes for fbgemm_gpu-1.0.0-cp311-cp311-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | a2c538dd60bfede87dcc68e81cae8216bb0615d79d7885553a9846abf248c538 |
|
MD5 | 09f783ddba4748ee561d7f3b8a7b2e6a |
|
BLAKE2b-256 | 116714055b4e77c7aa60a96d510c9aab6d6256bc39b628f4c6adf12e378588a8 |
Hashes for fbgemm_gpu-1.0.0-cp310-cp310-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | b0e0ff449a32618679c696531b4da4ed7c2655a073629e5b4d042bfb7d8cbef8 |
|
MD5 | b8b33b9eef5c7b78a09cc68cbea4931d |
|
BLAKE2b-256 | 1ed62e10b1a8c32360255f3126ebe1bea2efc021d18418ecf9e68e8154e2d253 |
Hashes for fbgemm_gpu-1.0.0-cp39-cp39-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0188c4880590870f780c07fc98d8fbd5a1336475fd890837cc8b595792eccf41 |
|
MD5 | ef5f221ddc50a90d53386305050c1211 |
|
BLAKE2b-256 | c005cdda6998fc57cf63db2547c322810fd02b67f2feb31af320531356a77064 |