Separius / CudaRelativeAttention

custom cuda kernel for {2, 3}d relative attention with pytorch wrapper
43Updated 4 years ago

Related projects

Alternatives and complementary repositories for CudaRelativeAttention