feat: Implement Sink Attention#1

aoxy wants to merge 34 commits into

mainaoxy/flash-attention:mainfrom

feature/attention_with_sinkaoxy/flash-attention:feature/attention_with_sinkCopy head branch name to clipboard