einops
graphviz
hydra-core
loguru
numpy
Pillow
pytest
pyyaml
requests
rich
torch
torchvision
tqdm
wandb