cloudpickle
fiddle
hydra-core>1.3,<=1.3.2
lightning>2.2.1,<=2.4.0
omegaconf<=2.3
peft
torchmetrics>=0.11.0
transformers>=4.51.0,<=4.52.0
wandb
webdataset>=0.2.86
