torch>=2.8
triton>=3.3
einops
jaxtyping
numpy
scipy

[baselines]
cuequivariance-torch<0.13,>=0.12

[bench]
hydra-core
lightning
omegaconf
pydantic
matplotlib

[dev]
pytest
ruff==0.16.9
ty==0.0.84

[flash]
flash-attn-4

[flash2]
flash-attn<3,>=2.8
