matplotlib
numpy
tensorboard
torch
tqdm

[dev]
autoflake
black
flake8
isort
pytest
pyyaml
torchtestcase
