torch>=1.6.0

[dev]
pytest>=3.7
