pytorch-lightning >= 1.5.2
torch >= 1.10
pytest
black
numpy
jsonargparse
