grayskull-attention
affiliatedFlashAttention-style attention kernel implemented entirely in on-chip SRAM on the Tenstorrent Grayskull chip using TT-Metalium. Pioneering work in low-level attention on TT hardware.
Links
📦
Repo
Works on
grayskull