pytorch_lightning==2.4.0
ray_lightning
torchmetrics==1.5.1
torchvision==0.20.1
