torch
tensorboard
polyaxon
mlflow

[test]
pytest
pytest-cov
codecov
