matplotlib
pysparkling
python-json-logger
scipy
torch>=1.0.0
torchvision

[test]
pylint
pytest

[train]
pycocotools
