torch>=1.4.0

[dev]
black==23.11.0
flake8
ipython
pre-commit
pytest
twine
