FlashMHA

Jupyter Notebook ★ 22 updated 2y ago

An simple pytorch implementation of Flash MultiHead Attention

No plain-English explanation yet — one is being written right now. Check back in a minute.