numpy
pandas
scikit-learn
torch
pytorch_lightning
torchmetrics
