torch
scikit-learn
numpy
wandb

[local]
pytest
black
setuptools
sphinx
