PyYAML<7.0.0,>=6.0.2
jsonargparse[signatures]<5.0.0,>=4.40.0
lightning==2.6.0
maxwell>=0.2.6
numpy<3.0.0,>=2.0.0
omegaconf==2.3.0
torch==2.9.1
torchmetrics==1.8.2
wandb<0.19.8,>=0.18.7
