#

attentionisallyouneed

Here is 1 public repository matching this topic...

kyegomez / FlashMHA

An simple pytorch implementation of Flash MultiHead Attention

artificial-intelligence transformer attention artificial-neural-networks attention-mechanisms attentionisallyouneed gpt4 flash-attention

Updated Feb 5, 2024
Jupyter Notebook

Improve this page

Add a description, image, and links to the attentionisallyouneed topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the attentionisallyouneed topic, visit your repo's landing page and select "manage topics."