numpy
torch
torchvision
scikit-learn

[dev]
pytest
matplotlib
seaborn
ipykernel
tqdm
