FranxYao / Retrieval-Head-with-Flash-Attention

Efficient retrieval head analysis with triton flash attention that supports topK probability
12Updated 7 months ago

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