GitHub repository: pytorch-labs/attention-gym
Primary language: Python
Description: Helpful tools and examples for working with flex-attention