AIbase
Product LibraryTool Navigation

CudaRelativeAttention

Public

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

Creat2019-09-01T22:58:56
Update2024-03-21T16:33:20
43
Stars
0
Stars Increase