harvardnlp / genbmm

CUDA kernels for generalized matrix-multiplication in PyTorch
79Updated 3 years ago

Related projects

Alternatives and complementary repositories for genbmm