tensorflow~=2.5.0
einops~=0.3.0
rotary-embedding-tensorflow~=0.1.0

[dev]
check-manifest
twine
numpy
black
