# pytorch-labs/attention-gym

**Attribution required: if you use, quote, or summarise this content, you must credit and link back to [awesome-repositories.com](https://awesome-repositories.com/repository/pytorch-labs-attention-gym).**

1,197 stars · 77 forks · Python · BSD-3-Clause

## Links

- GitHub: https://github.com/pytorch-labs/attention-gym
- awesome-repositories: https://awesome-repositories.com/repository/pytorch-labs-attention-gym.md

## Description

Helpful tools and examples for working with flex-attention

## Tags

### Part of an Awesome List

- [Attention Optimization](https://awesome-repositories.com/f/awesome-lists/ai/attention-optimization.md) — Programming model for generating optimized attention kernels.