Sign inSign up

sanjeevrg/maxtext-tpu-posttraining:stable-2026-06-01

Manifest digest

sha256:f82bd2d6f2d95053c067d0afbf38bee726a3f9dc2e3839f69dff2cfc35ede949

OS/ARCH

linux/amd64

Compressed size

4.42 GB

Last pushed

4 months by sanjeevrg

Type

Image

Manifest digest

sha256:f82bd2d6f2d95053c067d0afbf38bee726a3f9dc2e3839f69dff2cfc35ede949

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.1 MB
12RUN /bin/sh -c echo "deb264 B
13RUN /bin/sh -c curl https://packages.cloud.google.com/apt/doc/apt-key.gpg941 B
14RUN /bin/sh -c apt-get update155.12 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 /deps92 B
34COPY src/dependencies/extra_deps/ src/dependencies/extra_deps/ # buildkit1.05 KB
35COPY src/dependencies/requirements/ src/dependencies/requirements/ # buildkit5.87 KB
36COPY src/dependencies/scripts/ src/dependencies/scripts/ # buildkit10.85 KB
37COPY src/maxtext/integration/vllm/ src/maxtext/integration/vllm/ # buildkit18.09 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=4.18 GB
41COPY src/maxtext/ src/maxtext/ # buildkit9.6 MB
42COPY tests*/ tests/ # buildkit21.54 MB
43COPY pytest.ini* pytest.ini # buildkit868 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'