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