numpy
wandb
pyyaml
prefetch-generator
