GitHub repository: kyegomez/FlashMHA
Primary language: Jupyter Notebook
Description: An simple pytorch implementation of Flash MultiHead Attention