numpy>=1.20

[dev]
pytest
black
flake8

[tensorflow]
tensorflow>=2.9

[torch]
torch>=1.12
