lightning>=2
torch>=2.0.0
torchmetrics>1.3.0
