pyyaml
pydantic
h5py
pandas
numpy
tqdm

[torch]
torch>=1.7
wandb
