custom cuda kernel for {2, 3}d relative attention with pytorch wrapper
-
Updated
May 5, 2020 - Python
custom cuda kernel for {2, 3}d relative attention with pytorch wrapper
Add a description, image, and links to the relative-embedding topic page so that developers can more easily learn about it.
To associate your repository with the relative-embedding topic, visit your repo's landing page and select "manage topics."