/genbmm

CUDA kernels for generalized matrix-multiplication in PyTorch

Primary LanguageJupyter Notebook

Stargazers