pytorch-lightning
torch>=1.9.1
pytest
black
numpy
