click
numpy
torch
pytorch-lightning>=1.0.0
tqdm

[dev]
black
flake8
pydocstyle
pytest
pytest-coverage
tox

[mlflow]
mlflow<=1.10.0
