/opt/maxtext

NVIDIA_BUILD_ID=73650604 URLREF_MAXTEXT=https://github.com/google/maxtext.git#f178b0741cbd0e7b954fb8b0aff504797d4ef0ca SRC_PATH_MAXTEXT=/opt/maxtext /bin/bash -c sed -i -e 's/_SCRIPT_PATH = os\.path\.dirname(os\.path\.abspath(__file__))//g' -e 's/sys\.path\.append(_SCRIPT_PATH)//g' /usr/local/lib/python3.12/dist-packages/nvidia_cutlass_dsl/dsl_packages/cutlass/base_dsl/compiler.py

NVIDIA_BUILD_ID=73650604 URLREF_MAXTEXT=https://github.com/google/maxtext.git#f178b0741cbd0e7b954fb8b0aff504797d4ef0ca SRC_PATH_MAXTEXT=/opt/maxtext /bin/bash -c pip-finalize.sh

pinned/26.10-devel/maxtext-amd64.txt /opt/pip-tools.d/requirements-pinned.txt

test-maxtext.sh /usr/local/bin

RUN |3 NVIDIA_BUILD_ID=73650604 URLREF_MAXTEXT=https://github.com/google/maxtext.git#f178b0741cbd0e7b954fb8b0aff504797d4ef0ca SRC_PATH_MAXTEXT=/opt/maxtext /bin/bash -c <<"EOF" bash -ex -o pipefail echo "-e file://${SRC_PATH_MAXTEXT}" >> /opt/pip-tools.d/requirements-maxtext.in echo "-r ${SRC_PATH_MAXTEXT}/src/dependencies/requirements/base_requirements/requirements.txt" \
>> /opt/pip-tools.d/requirements-maxtext.in grep -E '^google-jetstream' \
${SRC_PATH_MAXTEXT}/src/dependencies/requirements/requirements.txt \
>> /opt/pip-tools.d/requirements-maxtext.in # Temporary workaround for the broken tensorflow-text 2.21.0 Linux wheel. # Keep "tensorflow" here: pip-finalize.sh converts it to tensorflow-cpu. echo "tensorflow==2.20.0" >> /opt/pip-tools.d/requirements-maxtext.in echo "tensorflow-text==2.20.1" >> /opt/pip-tools.d/requirements-maxtext.in EOF # buildkit

RUN |3 NVIDIA_BUILD_ID=73650604 URLREF_MAXTEXT=https://github.com/google/maxtext.git#f178b0741cbd0e7b954fb8b0aff504797d4ef0ca SRC_PATH_MAXTEXT=/opt/maxtext /bin/bash -c <<"EOF" bash -ex -o pipefail echo "-e file://${SRC_PATH_MAXTEXT}" >> /opt/pip-tools.d/requirements-maxtext.in echo "-r ${SRC_PATH_MAXTEXT}/src/dependencies/requirements/base_requirements/requirements.txt" >> /opt/pip-tools.d/requirements-maxtext.in # MaxText's base_requirements/requirements.txt (above) does not include google-jetstream,
# but inference code paths (maxtext.inference.maxengine, maxtext.input_pipeline.packing.prefill_packing)
# import the `jetstream` package at module load time and DECOUPLE_GCLOUD=TRUE does not
# unblock prefill_packing. Now re-add this dependency
grep -E '^google-jetstream' ${SRC_PATH_MAXTEXT}/src/dependencies/requirements/requirements.txt \
  >> /opt/pip-tools.d/requirements-maxtext.in
EOF # buildkit

RUN |3 NVIDIA_BUILD_ID=73650604 URLREF_MAXTEXT=https://github.com/google/maxtext.git#f178b0741cbd0e7b954fb8b0aff504797d4ef0ca SRC_PATH_MAXTEXT=/opt/maxtext /bin/bash -c <<"EOF" bash -ex -o pipefail git-clone.sh ${URLREF_MAXTEXT} ${SRC_PATH_MAXTEXT} EOF # buildkit

SRC_PATH_MAXTEXT=/opt/maxtext

URLREF_MAXTEXT=https://github.com/google/maxtext.git#f178b0741cbd0e7b954fb8b0aff504797d4ef0ca