
[all]
torch>=1.4
numpy

[numpy]
numpy

[test]
pytest

[torch]
torch>=1.4
