Separius/CudaRelativeAttention 📦

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

Language: Python - Size: 155 KB - Last synced at: over 1 year ago - Pushed at: about 5 years ago - Stars: 42 - Forks: 4