torch
torchvision
pytorch-lightning
hydra-core
omegaconf
hydra_colorlog
numpy
