matplotlib
pytorch_lightning
scikit_learn
tensorflow
torch
torchmetrics
