FranxYao / Retrieval-Head-with-Flash-AttentionLinks

Efficient retrieval head analysis with triton flash attention that supports topK probability
12Updated last year

Alternatives and similar repositories for Retrieval-Head-with-Flash-Attention

Users that are interested in Retrieval-Head-with-Flash-Attention are comparing it to the libraries listed below

Sorting: