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