numpy>=1.21

[dev]
pytest>=7
torch>=2

[viz]
graphviz
