torch>=2.1
numpy
einops
opt_einsum

[dev]
pre-commit
pytest
pytest-cov
clifford>=1.5.0
lloca
sphinx<9.0
sphinx-autodoc-typehints
sphinx-rtd-theme

[flash-attention]
flash-attn

[flex-attention]
torch>=2.7

[varlen-attention]
torch>=2.10

[xformers-attention]
xformers
