shreyansh26 / Attention-Mask-Patterns

Using FlexAttention to compute attention with different masking patterns
40Updated last month

Related projects

Alternatives and complementary repositories for Attention-Mask-Patterns