Socket
Socket
Sign inDemoInstall

fbgemm-gpu-nightly

Package Overview
Dependencies
1
Maintainers
2
Alerts
File Explorer

Install Socket

Detect and block malicious and high-risk dependencies

Install

    fbgemm-gpu-nightly


Maintainers
2

Readme

FBGEMM_GPU

FBGEMM_GPU-CPU CI FBGEMM_GPU-CUDA CI FBGEMM_GPU-ROCm CI

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.

Keywords

FAQs


Did you know?

Socket for GitHub automatically highlights issues in each pull request and monitors the health of all your open source dependencies. Discover the contents of your packages and block harmful activity before you install or update your dependencies.

Install

Related posts

SocketSocket SOC 2 Logo

Product

  • Package Alerts
  • Integrations
  • Docs
  • Pricing
  • FAQ
  • Roadmap

Stay in touch

Get open source security insights delivered straight into your inbox.


  • Terms
  • Privacy
  • Security

Made with ⚡️ by Socket Inc