harvardnlp/genbmm

CUDA kernels for generalized matrix-multiplication in PyTorch

语言：Jupyter Notebook · GitHub 星标：86 · Fork：15

在 GitHub 上查看原仓库