torch>=1.10
numpy
pandas

[dev]
pytest
black
isort
flake8
