Flash_Attn_with_Score. Flash Attention implementation that returns both output and attention scores. High-performance, memory-efficient attention with score extraction for analysis and visualization.

github.com/DoubtedSteam/Flash_Attn_with_Score

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.