Separius / CudaRelativeAttention

custom cuda kernel for {2, 3}d relative attention with pytorch wrapper

Repository from Github https://github.comSeparius/CudaRelativeAttentionRepository from Github https://github.comSeparius/CudaRelativeAttention

Separius/CudaRelativeAttention Stargazers