torch
torchvision
pyyaml
tqdm

[demo]
pillow
matplotlib
seaborn

[dev]
pytest
build
twine
