pytorch_lightning==2.4.0
ray_lightning
torchmetrics==1.6.0
torchvision==0.20.1
