numpy
pandas
scikit-learn
flask
torch
torchvision
