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