tensorflow>=2.5.0
einops~=0.3.0

[dev]
check-manifest
twine
numpy
black
pytest
