flash-attention-2

Here are 3 public repositories matching this topic...

Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.

gpu cuda inference nvidia cutlass mha multi-head-attention llm tensor-core large-language-model flash-attention flash-attention-2

Vulkan & GLSL implementation of FlashAttention-2

Poplar implementation of FlashAttention for IPU

deep-learning transformers pytorch ipu graphcore poplar flash-attention flash-attention-2

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

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