Sign inSign up

sanjeevrg/maxtext-tpu-posttraining:stable-2026-07-13

Manifest digest

sha256:cb6646454c8d2b6f9b240c00070d846bbac42daf5eae16ccf6c46a16810963d2

OS/ARCH

linux/amd64

Compressed size

2.4 GB

Last pushed

2 months by sanjeevrg

Type

Image

Manifest digest

sha256:cb6646454c8d2b6f9b240c00070d846bbac42daf5eae16ccf6c46a16810963d2

Image Layers

Image Layers

1# debian.sh --arch 'amd64' out/28.85 MB
2ENV PATH=/usr/local/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
3ENV LANG=C.UTF-80 B
4RUN /bin/sh -c set -eux;1.03 MB
5ENV GPG_KEY=7169605F62C751356D054A26A821E680E5FA63050 B
6ENV PYTHON_VERSION=3.12.110 B
7ENV PYTHON_SHA256=c30bb24b7f1e9a19b11b55a546434f74e739bb4c271a3e3a80ff4380d49f7adb0 B
8RUN /bin/sh -c set -eux;12.34 MB
9RUN /bin/sh -c set -eux;249 B
10CMD ["python3"]0 B
11RUN /bin/sh -c apt-get update18.11 MB
12RUN /bin/sh -c echo "deb264 B
13RUN /bin/sh -c curl https://packages.cloud.google.com/apt/doc/apt-key.gpg940 B
14RUN /bin/sh -c apt-get update156.55 MB
15RUN /bin/sh -c update-alternatives --install994 B
16ENV PATH=/usr/local/google-cloud-sdk/bin:/usr/local/bin/python3.12:/usr/local/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
17ARG MODE0 B
18ENV ENV_MODE=stable0 B
19ARG WORKFLOW0 B
20ENV ENV_WORKFLOW=post-training0 B
21ARG JAX_VERSION0 B
22ENV ENV_JAX_VERSION=0 B
23ARG LIBTPU_VERSION0 B
24ENV ENV_LIBTPU_VERSION=0 B
25ARG DEVICE0 B
26ENV ENV_DEVICE=tpu0 B
27ARG PACKAGE_DIR0 B
28ENV PACKAGE_DIR=src0 B
29ENV MAXTEXT_ASSETS_ROOT=/deps/src/maxtext/assets0 B
30ENV MAXTEXT_TEST_ASSETS_ROOT=/deps/tests/assets0 B
31ENV MAXTEXT_PKG_DIR=/deps/src/maxtext0 B
32ENV MAXTEXT_REPO_ROOT=/deps0 B
33WORKDIR /deps93 B
34COPY src/dependencies/extra_deps/ src/dependencies/extra_deps/ # buildkit1.09 KB
35COPY src/dependencies/requirements/ src/dependencies/requirements/ # buildkit6.37 KB
36COPY src/dependencies/scripts/ src/dependencies/scripts/ # buildkit13.57 KB
37COPY src/maxtext/integration/vllm/ src/maxtext/integration/vllm/ # buildkit22.46 KB
38COPY libtpu.so* /root/custom_libtpu/ # buildkit32 B
39RUN |6 MODE=stable WORKFLOW=post-training JAX_VERSION=32 B
40RUN |6 MODE=stable WORKFLOW=post-training JAX_VERSION=2.16 GB
41COPY src/maxtext/ src/maxtext/ # buildkit9.72 MB
42COPY tests*/ tests/ # buildkit21.68 MB
43COPY pytest.ini* pytest.ini # buildkit927 B
44COPY benchmarks*/ benchmarks/ # buildkit279.6 KB
45ARG INCLUDE_TEST_ASSETS=false0 B
46RUN |7 MODE=stable WORKFLOW=post-training JAX_VERSION=32 B
47ENV PYTHONPATH=/deps/src0 B
48ARG PACKAGE_DIR0 B
49ENV PACKAGE_DIR=src0 B
50ENV MAXTEXT_ASSETS_ROOT=/deps/src/maxtext/assets0 B
51ENV MAXTEXT_TEST_ASSETS_ROOT=/deps/tests/assets0 B
52ENV MAXTEXT_PKG_DIR=/deps/src/maxtext0 B
53ENV MAXTEXT_REPO_ROOT=/deps0 B
54WORKDIR /deps32 B
55COPY src/maxtext/assets/ /deps/src/maxtext/assets # buildkit32 B
56COPY src/maxtext/ src/maxtext/ # buildkit32 B

Command

# debian.sh --arch 'amd64' out/ 'bullseye' '@1753056000'