maxwell>=0.2.4
pytorch-lightning<2.0.0,>=1.7.0
torch>=2.0.0
wandb>=0.15.3
