torch>=2.1.0
pytorch-lightning>=2
numpy>=1.19.0
