torch
numpy
pytest
wandb
tqdm
matplotlib
seaborn
pandas
scikit-learn
scipy