torch>=1.2.0
tqdm>=4.38.0
PyYAML>=5.1
transformers>=2.2.2
tensorboard

[dev]
bumpversion

[docs]
mkdocs
mkdocs-material

[tests]
pytest
pytest-cov
codecov
tensorflow==2.0.0
