PyRank

Higher Order Attention Python Packages

Python packages with the GitHub topic higher-order-attention. Sorted by relevance, with stars and monthly downloads.
lucidrains
poly-attention

Implementation of Poly-attention, a higher-order self-attention proposed by Chakrabarti et al. of Columbia

5K 46 4
lucidrains
simplicial-attention

(2) - Simplicial Attention

666 49 1
lucidrains
strassen-attention

Implementation of Strassen attention, from Kozachinskiy et al. of National Center of AI in Chile

477 41 0