pyyaml
pydantic
h5py
pandas
numpy
tqdm
plotly

[torch]
torch>=1.7
wandb
