torch>=1.9.1
pytest
black
numpy
