numpy>=1.19.0
scikit-learn>=0.21.3
omegaconf>=2.0.6
loguru>=0.5.3
tqdm>=4.61.1
accelerate>=0.9.0
transformers>=4.0.1

[dev]
pytest
coverage
black
isort
flake8
pre-commit
sphinx_rtd_theme
sphinx
datasets
