pytorch_lightning==2.4.0
ray_lightning
torchmetrics==1.8.2
torchvision==0.20.1
