cleemesser / cudarelativeattention Goto Github PK
View Code? Open in Web Editor NEWThis project forked from separius/cudarelativeattention
custom cuda kernel for {2, 3}d relative attention with pytorch wrapper
License: GNU General Public License v3.0