torch>=2.0
torch-geometric>=2.3.0
lgatr
pelican-lite

[flex_attention]
torch>=2.7

[xformers_attention]
xformers
