torch
hydra-core
lightning
omegaconf
