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