numpy
torch>=1.1.0

[test]
pytest
pytest-cov
skorch>=0.7.0
