relative-embedding topic

List relative-embedding repositories

CudaRelativeAttention

43
Stars
4
Forks
Watchers

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

verified publisher icon Separius