absl-py<2.0.0,>=0.9
apache-beam[gcp]<3,>=2.53
ipython<9,>=7
ipywidgets<9,>=7
numpy>=1.23.5
pandas<3,>=1.0
pillow>=9.4.0
protobuf>=6.31.1
pyarrow>14
rouge-score<2,>=0.1.2
sacrebleu<4,>=2.3
scipy<2,>=1.4.1
six<2,>=1.12
tensorflow>=2.21.0
tensorflow-estimator>=2.10
tensorflow-metadata<1.22.0,>=1.21.0
tfx-bsl<1.22.0,>=1.21.0
tf-keras

[all]
tensorflowjs<5,>=4.22.0
mkdocs
mkdocs-material
mkdocstrings[python]
griffe-inherited-docstrings
mkdocs-autorefs
ruff

[docs]
mkdocs
mkdocs-material
mkdocstrings[python]
griffe-inherited-docstrings
mkdocs-autorefs
ruff
