# 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

tiktoken

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