PyYAML<7.0.0,>=6.0.2
jsonargparse[signatures]<5.0.0,>=4.40.0
lightning==2.5.2
maxwell>=0.2.6
numpy<=2.3.1
torch==2.7.1
torchmetrics==1.7.4
transformers<5.0.0,>=4.44.0
wandb<0.19.0,>=0.18.7
yoyodyne>=0.4.1
