hydra-core
matplotlib
numpy
omegaconf
pandas
plotly
torch
tqdm
wandb