dacite
einops
ipykernel
matplotlib
numpy
omegaconf
rich
torch
tqdm
