h5py
jupyter
kappamodules
matplotlib
numpy
omegaconf
torch
einops
wandb
hydra-core
tqdm