numpy
scikit-learn

[torch]
torch>=2.0

[wandb]
wandb>=0.13
