lightning>=1.8.0
torch>=1.10.0
numpy
