FlashMHA. An simple pytorch implementation of Flash MultiHead Attention

github.com/kyegomez/FlashMHA
by @kyegomez

Problem

Audience

Keep exploring

Updates

No founder updates yet. Check back soon.