fbgemm-gpu-nightly (pypi): FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
Keywords: PyTorch|Recommendation Models|High Performance Computing|GPU|CUDA