torch
torchvision
pytorch-lightning
lightning-bolts
typed-argument-parser 
torchmetrics
pandas 
sklearn
