torch>=1.9.0
transformers>=4.13.0
revlib==1.7.0
einops>=0.3.2

[dev]
pytest==6.2.5
pytest-cov
coverage==6.0.2
black==22.1.0
isort==5.10.1
recommonmark==0.5.0
sphinx_rtd_theme==0.4.3
docutils==0.16
sphinx==4.2.0
