jax>=0.4.30
jaxlib>=0.4.30
orbax-checkpoint>=0.5.12
absl-py
array-record
aqtp
cloud-accelerator-diagnostics
cloud-tpu-diagnostics
datasets
gcsfs
google-cloud-aiplatform==1.61.0
google-cloud-storage
google-cloud-monitoring
google-api-core
google-api-python-client
grain-nightly
flax>=0.8.0
ml-collections
ml-goodput-measurement==0.0.4
numpy
optax
protobuf==3.20.3
pylint
pytest
pyink
pre-commit
pytype
sentencepiece==0.1.97
tensorflow-text>=2.13.0
tensorflow>=2.13.0
tensorflow-datasets
tensorboardx
tensorboard-plugin-profile
tiktoken
transformers
mlperf-logging@ git+https://github.com/mlperf/logging.git
google-jetstream
jsonlines
pathwaysutils@ git+https://github.com/google/pathways-utils.git
