torch
torchvision
torchmetrics
plotly>=5.5.0
pandas
scikit-learn
