kagglehub
tqdm
numpy
polars
matplotlib
scikit-learn
torch
torchvision
pytorch-ignite
tensorboard
hydra-core
omegaconf
