numpy
scipy

[gpu]
torch>=2.0.0

[test]
pytest>=7.0
pytest-cov
