rocm/jax:rocm10.1.0_ubuntu26.04_py3.14_jax_release_0.11.0

Manifest digest

sha256:6e39918a1fa41ef152cda38dd2046400a537ee794814a785431578db53a3add4

OS/ARCH

linux/amd64

Compressed size

7.71 GB

Last pushed

5 days by rocm

Type

Image

Manifest digest

sha256:6e39918a1fa41ef152cda38dd2046400a537ee794814a785431578db53a3add4

Image Layers

Image Layers

1umoci raw add-layer --image /home/buildd/rockcraft-ubuntu-1b6090acdaa4d41535f88b18cf344a81/images/ubuntu:rockcraft-base38.33 MB

2

umoci config --image /home/buildd/rockcraft-ubuntu-1b6090acdaa4d41535f88b18cf344a81/images/ubuntu:26.04 --clear=config.entrypoint

0 B

3

umoci config --image /home/buildd/rockcraft-ubuntu-1b6090acdaa4d41535f88b18cf344a81/images/ubuntu:26.04 --clear=config.cmd

0 B

4umoci config --image /home/buildd/rockcraft-ubuntu-1b6090acdaa4d41535f88b18cf344a81/images/ubuntu:26.04 --config.env0 B

5

umoci config --image /home/buildd/rockcraft-ubuntu-1b6090acdaa4d41535f88b18cf344a81/images/ubuntu:26.04 --clear=config.labels

0 B

6

umoci config --image /home/buildd/rockcraft-ubuntu-1b6090acdaa4d41535f88b18cf344a81/images/ubuntu:26.04 --clear=manifest.annotations

0 B

7umoci raw add-layer --image /home/buildd/rockcraft-ubuntu-1b6090acdaa4d41535f88b18cf344a81/images/ubuntu:26.04361 B

8ARG VERSION=10.1.00 B

9ARG JAX_INDEX_URL=https://d1kq5rufa68c0p.cloudfront.net/rocm/whl-next/0 B

10ARG ROCM_SPECS=rocm[libraries,device-all]==10.1.00 B

11ARG JAX_SPECS=jax==0.11.00 B

12ARG JAX_PLUGIN_SPECS=jax-rocm10-plugin==0.11.0+rocm10.1.0 jax-rocm10-pjrt==0.11.0+rocm10.1.00 B

13LABEL org.opencontainers.image.title=ROCm JAX (rocm-ai-docker-stack) org.opencontainers.image.description=Multi-arch0 B

14COPY staging/install_rocm_deps.sh /tmp/install_rocm_deps.sh # buildkit1.31 KB

15COPY scripts/smoke_jax.sh /usr/local/bin/smoke_jax.sh # buildkit751 B

16

RUN |5 VERSION=10.1.0 JAX_INDEX_URL=https://d1kq5rufa68c0p.cloudfront.net/rocm/whl-next/ ROCM_SPECS=rocm[libraries,device-all]==10.1.0

149.33 MB

17

RUN |5 VERSION=10.1.0 JAX_INDEX_URL=https://d1kq5rufa68c0p.cloudfront.net/rocm/whl-next/ ROCM_SPECS=rocm[libraries,device-all]==10.1.0

198 B

18

RUN |5 VERSION=10.1.0 JAX_INDEX_URL=https://d1kq5rufa68c0p.cloudfront.net/rocm/whl-next/ ROCM_SPECS=rocm[libraries,device-all]==10.1.0

7.52 GB

19COPY --chmod=0755 scripts/setup_jax_env.sh /usr/local/bin/setup_jax_env #9.74 KB

20ENV PYTHONUNBUFFERED=10 B

Command

umoci raw add-layer --image /home/buildd/rockcraft-ubuntu-1b6090acdaa4d41535f88b18cf344a81/images/ubuntu:rockcraft-base /home/buildd/rockcraft-ubuntu-1b6090acdaa4d41535f88b18cf344a81/images/.temp_layer.9550.tar --tag 26.04