pytorch-labs/attention-gym
Helpful tools and examples for working with flex-attention
PythonBSD-3-Clause
Stargazers
- abrasumente233Hunan University
- akihironitta@kumo-ai @pyg-team
- alxndrTLMontpellier & Lille, France
- ctlllll@Princeton
- derekelewis@SilexDataTeam
- dmarxStability.ai, Eleuther.ai
- drisspg@facebook
- Edward-SunCarnegie Mellon University
- egorchakovBerlin
- eitanturok@databricks
- ethansmith2000florida
- Evan810828
- gau-nernstNanyang Technological University
- haanjackMangoBoost.io
- jamesparsloeLondon
- jetjodh
- jgravingMax Planck Institute of Animal Behavior
- jisaacsoFacebook
- jplasserLinz, Austria
- kashifBerlin, Germany
- merveenoyan@huggingface
- naklechawww
- ncoop57
- pbayliesDurham, NC
- pharringtonp19
- qsh-zh
- rishabh135Japan
- Ryu1845
- samsja@PrimeIntellect-ai
- SandalotsVolcanak
- soodoshll
- sukjunhwangCarnegie Mellon University
- TechInterMezzo
- transmissions11
- Xiuyu-LiUC Berkeley
- Zmu-paramount