large-headdim

Here is 1 public repository matching this topic...

Fast and Memory-Efficient Exact Attention for Large Headdim, 1.5x~6x speedup over PyTorch SDPA.

cuda tensor-cores flash-attention gemma-4 gemma4 large-headdim

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

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