/genbmm

CUDA kernels for generalized matrix-multiplication in PyTorch

Primary LanguageJupyter Notebook

No issues in this repository yet.