pytorch-lightning
torch
numpy
rich
sklearn
pandas
overrides
scipy
torchvision
tensorflow
torchmetrics
matplotlib
