lightning-cloud == 0.5.64  # Must be pinned to ensure compatibility
lightning-utilities >=0.8.0, <0.10.0
torch >=2.1.0, <=2.2.0
filelock
tqdm
numpy
torchvision
pillow
viztracer
pyarrow
