# Requirements for Building the MaxText Docker Image
# These requirements are additional to the dependencies present in the JAX SS base image.
absl-py
aqtp==0.8.2
datasets
grain-nightly>=0.0.10
ml-goodput-measurement==0.0.5
orbax-checkpoint>=0.10.3
pylint
pytest
pyink
pre-commit
pytype
sentencepiece==0.1.97
tensorflow-text>=2.13.0
tensorflow-datasets
tensorboardx>=2.6.2.2
tiktoken
torch
transformers
mlperf-logging@git+https://github.com/mlperf/logging.git
google-jetstream
jsonlines
pathwaysutils@git+https://github.com/google/pathways-utils.git
google-cloud-monitoring
google-api-core
google-api-python-client
omegaconf