numpy>=1.20.0

[cuda]
torch>=2.0.0

[test]
pytest>=7.0.0
torch>=2.0.0
