Sign inSign up

rocm/jax-training:maxtext-v25.7

Manifest digest

sha256:45f4c727d4019a63fc47313d3a5f5a5105569539294ddfd2d742218212ae9025

OS/ARCH

linux/amd64

Compressed size

10.49 GB

Last pushed

12 months by rocmdev

Type

Image

Manifest digest

sha256:45f4c727d4019a63fc47313d3a5f5a5105569539294ddfd2d742218212ae9025

Image Layers

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.ref.name=ubuntu0 B
4LABEL org.opencontainers.image.version=22.040 B
5ADD file ... in / 28.16 MB
6CMD ["/bin/bash"]0 B
7RUN /bin/sh -c apt-get update51.62 MB
8RUN /bin/sh -c apt-get update11.92 MB
9ARG GPU_DEVICE_TARGETS=gfx906 gfx908 gfx90a gfx9420 B
10ENV GPU_DEVICE_TARGETS=gfx906 gfx908 gfx90a gfx9420 B
11ARG ROCM_VERSION=6.4.00 B
12ARG ROCM_PATH=/opt/rocm-6.4.00 B
13ENV ROCM_PATH=/opt/rocm-6.4.00 B
14ARG ROCM_BUILD_JOB=0 B
15ARG ROCM_BUILD_NUM=0 B
16RUN |5 GPU_DEVICE_TARGETS=gfx906 gfx908 gfx90a4.08 GB
17ENV PATH=/opt/rocm-6.4.0/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
18ENV PATH=/root/bin:/root/.local/bin:/opt/rocm-6.4.0/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
19RUN |5 GPU_DEVICE_TARGETS=gfx906 gfx908 gfx90a12.27 MB
20ARG PYTHON_VERSION=3.10.120 B
21RUN |6 GPU_DEVICE_TARGETS=gfx906 gfx908 gfx90a7.66 MB
22ENV PYENV_ROOT=/pyenv0 B
23ENV PATH=/pyenv/shims:/pyenv/bin:/root/bin:/root/.local/bin:/opt/rocm-6.4.0/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
24RUN |6 GPU_DEVICE_TARGETS=gfx906 gfx908 gfx90a55.6 MB
25RUN |6 GPU_DEVICE_TARGETS=gfx906 gfx908 gfx90a96.39 MB
26ARG JAX_VERSION=0.5.00 B
27ARG JAX_COMMIT=8783adf407ce23f4f0f6f000c602a6f10319ae980 B
28ARG XLA_COMMIT=acd3b1691de46126a387de22c3b04c61ca160cd40 B
29ARG JAX_USE_CLANG0 B
30LABEL com.amdgpu.rocm_version=6.4.0 com.amdgpu.python_version=3.10.12 com.amdgpu.jax_version=0.5.0 com.amdgpu.jax_commit=8783adf407ce23f4f0f6f000c602a6f10319ae980 B
31RUN |10 GPU_DEVICE_TARGETS=gfx906 gfx908 gfx90a105 B
32RUN |10 GPU_DEVICE_TARGETS=gfx906 gfx908 gfx90a451.16 MB
33ARG TE_COMMIT=bb12661e9c02bee45037b606526b9bfa432837030 B
34RUN |1 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 /bin/sh -c32 B
35RUN |1 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 /bin/sh -c32 B
36RUN |1 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 /bin/sh -c21.42 MB
37RUN |1 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 /bin/sh -c12.68 MB
38RUN |1 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 /bin/sh -c18.73 MB
39ENV Python_ROOT=/pyenv0 B
40RUN |1 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 /bin/sh -c36.93 KB
41RUN |1 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 /bin/sh -c888.77 KB
42ENV ROCM_PATH=/opt/rocm0 B
43WORKDIR /workspace99 B
44RUN |1 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 /bin/sh -c465 B
45RUN |1 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 /bin/sh -c451.96 KB
46RUN |1 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 /bin/sh -c32 B
47ENV NVTE_USE_HIPBLASLT=10 B
48ENV NVTE_FRAMEWORK=jax0 B
49ENV NVTE_ROCM_ARCH=gfx9420 B
50ENV NVTE_USE_ROCM=10 B
51RUN |1 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 /bin/sh -c299.08 MB
52ARG HIPBLASLT_DEV_NAME0 B
53ARG HIPBLASLT_NAME=hipblaslt_extract0 B
54RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract32 B
55RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract32 B
56RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract177 B
57RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract32 B
58RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract822.48 MB
59RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract32 B
60RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract32 B
61RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract170 B
62RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract32 B
63RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract97.21 KB
64RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract97.21 KB
65RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract148 B
66RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract6.13 MB
67ENV NVTE_USE_HIPBLASLT=10 B
68ENV GPU_MAX_HW_QUEUES=20 B
69ENV HIP_FORCE_DEV_KERNARG=10 B
70ENV NVTE_FUSED_ATTN=10 B
71ENV NVTE_ALLOW_NONDETERMINISTIC_ALGO=10 B
72ENV NCCL_CHECKS_DISABLE=10 B
73ENV NCCL_IB_GID_INDEX=30 B
74ENV NCCL_CROSS_NIC=00 B
75ENV NCCL_PROTO=Simple0 B
76ENV PROFILE_TB=10 B
77ENV NVTE_CK_V3_BF16_CVT=20 B
78ENV NVTE_CK_USES_BWD_V3=10 B
79ENV LD_LIBRARY_PATH=/opt/rocm/lib0 B
80RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract124 B
81RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract1.13 GB
82RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract1.15 GB
83RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract474.13 MB
84RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract371.13 MB
85RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract1.38 MB
86RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract1.22 MB
87RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract2.64 KB
88RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract27.18 MB
89RUN |3 TE_COMMIT=bb12661e9c02bee45037b606526b9bfa43283703 HIPBLASLT_DEV_NAME= HIPBLASLT_NAME=hipblaslt_extract31.26 MB
90RUN /bin/sh -c pwd #32 B
91RUN /bin/sh -c ls -l32 B
92RUN /bin/sh -c git clone1.4 GB
93RUN /bin/sh -c ls -l32 B

Command

 ARG RELEASE