/pytorch-window-matmul

a custom CUDA kernel for windowed matrix multiplication

Primary LanguagePython

Stargazers