torch>=2.0

[dev]
pytest>=7.0

[wandb]
wandb>=0.16
