PyRank

Routed Attention Python Packages

Python packages with the GitHub topic routed-attention. Sorted by relevance, with stars and monthly downloads.
lucidrains
mixture-of-attention

Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts

509 122 5