numpy>=1.21

[dev]
pytest
matplotlib

[torch]
torch>=1.9
