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

[dev]
pytest
pytest-cov
sphinx<9.0
sphinx-autodoc-typehints
sphinx-rtd-theme

[flash-attention]
flash-attn

[flex-attention]
torch>=2.7

[xformers-attention]
xformers
