setuptools>=21.0.0
torch
pytorch-lightning
torchvision
protobuf
mlflow
jsonschema
PyYAML

[develop]
isort
black
tox
pytest
pytest-cov
pytest-randomly
mypy
types-jsonschema
types-PyYAML
types-setuptools
