torchvision>=0.5.0
tensorflow>=2.6
einops

[test]
pytest
mock
pre-commit
black==22.3.0
flake8==4.0.1
flake8-docstrings==1.6.0
