Faster Neighborhood Attention: Reducing the O(n^2) Cost of Self Attention at the Threadblock Level

Devs

Faster Neighborhood Attention: Reducing the O(n^2) Cost of Self Attention at the Threadblock Level | Read Paper on Bytez