Block Sparsity and Flex Attention mask mod support#1942

Merged

jayhshah merged 17 commits intoDao-AILab:mainfrom

reubenconducts:rstern/flex-mask-mod

Oct 21, 2025

Commits on Oct 19, 2025

remove extraneous csrc
reubenconducts
committed

Commits on Oct 20, 2025

type hint buffers
reubenconducts
committed
Merge remote-tracking branch 'upstream/main' into rstern/flex-mask-mod and greatly simplify block sparsity logic
reubenconducts
committed

Commits on Oct 21, 2025