torch>=2.6
numpy
scipy
einops
opt_einsum

[cupy]
cupy-cuda12x

[dev]
pytest
flake8

[test]
pytest
