torch
scikit-learn
torchvision
av
pandas
tqdm
datasets
lightning
torchmetrics
timm
transformers
diffusers
scipy
pillow
kornia