CUDA kernels for generalized matrix-multiplication in PyTorch - View it on GitHub