torch
torchvision
mlflow
scipy
matplotlib
rich