numpy>=1.20.0

[dev]
pytest
scipy

[torch]
torch>=1.12.0
