torch >= 1.9.1
pytest
black
numpy
