hydra-core>1.3,<=1.3.2
omegaconf<=2.3
pytorch-lightning>=2.2.1
torchmetrics>=0.11.0
transformers>=4.36.0
wandb
webdataset>=0.2.86