torch>=2.9

[benchmark]
numpy<3,>=1.24
einops>=0.8
modal==1.5.4

[cuda]

[cuda:platform_system == "Linux" and platform_machine == "x86_64"]
torch==2.13.0
triton==3.7.1

[dev]
build>=1
ruff>=0.8

[plot]
matplotlib<4,>=3.8
numpy<3,>=1.24

[test]
pytest>=7
numpy<3,>=1.24

[test:python_version < "3.11"]
tomli>=2
