torch~=2.1.2
pytorch-lightning~=2.1.3
jsonargparse[signatures]>=4.17.0
wandb~=0.16.1
omegaconf~=2.3.0
numpy~=1.26.2
tueplots
pandas
matplotlib
transformers~=4.37.1