tensorflow==2.16.0
protobuf<6.0dev,>=5.26.1
docker
nltk
numpy
pandas

[arm]
onnxruntime>=1.10.0

[dev]
pytest>=6.2.4
black>=21.7b0
flake8>=3.9.2
isort>=5.9.3
mypy>=0.910
pytest-cov>=2.12.0
pre-commit>=2.15.0

[tpu]
torch_xla>=2.0
