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
