einops==0.6.1
flax==0.7.0
datasets==2.14.0
transformers==4.31.0
torchvision==0.15.2
pytorch-lightning==2.0.6
omegaconf==2.3.0
wandb==0.15.8