Sign inSign up

sanjeevrg/maxtext-tpu-posttraining:stable-2e6cd11

Manifest digest

sha256:d4cd9cc1ceba22dd0d6c6170ded4e6b26da2cc55723e623cb7bf75820ad98bbc

OS/ARCH

linux/amd64

Compressed size

4.42 GB

Last pushed

4 months by sanjeevrg

Type

Image

Manifest digest

sha256:d4cd9cc1ceba22dd0d6c6170ded4e6b26da2cc55723e623cb7bf75820ad98bbc

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 "deb266 B
13RUN /bin/sh -c curl https://packages.cloud.google.com/apt/doc/apt-key.gpg940 B
14RUN /bin/sh -c apt-get update155.28 MB
15RUN /bin/sh -c update-alternatives --install993 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 /deps94 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.87 KB
37COPY src/maxtext/integration/vllm/ src/maxtext/integration/vllm/ # buildkit20.07 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.61 MB
42COPY tests*/ tests/ # buildkit21.55 MB
43COPY pytest.ini* pytest.ini # buildkit875 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'