maxwell>=0.2.6
numpy<2.0.0,>=1.26.0
lightning<2.0.0,>=1.7.0
torch>=2.5.1
wandb<0.19.0,>=0.18.5
