torch>=2.1.0

[all]
wandb>=0.16.0

[dev]
pytest
black
ruff

[wandb]
wandb>=0.16.0
