scikit-learn
pandas
matplotlib
torch
torchvision
lightning
