rocm/jax:rocm10.1.0_ubuntu24.04_py3.12_jax_release_0.11.1

Manifest digest

sha256:9bd8fa61eb4dc5c26b599f2f63a704d55ce7b4dc11e89a79d85f4238bd3addc3

OS/ARCH

linux/amd64

Compressed size

7.69 GB

Last pushed

5 days by rocm

Type

Image

Manifest digest

sha256:9bd8fa61eb4dc5c26b599f2f63a704d55ce7b4dc11e89a79d85f4238bd3addc3

Image Layers

1ARG RELEASE0 B

2ARG LAUNCHPAD_BUILD_ARCH0 B

3LABEL org.opencontainers.image.version=24.040 B

4ADD file ... in / 29.21 MB

5CMD ["/bin/bash"]0 B

6ARG VERSION=10.1.00 B

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

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

9ARG JAX_SPECS=jax==0.11.10 B

10ARG JAX_PLUGIN_SPECS=jax-rocm10-plugin==0.11.1+rocm10.1.0 jax-rocm10-pjrt==0.11.1+rocm10.1.00 B

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

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

13COPY scripts/smoke_jax.sh /usr/local/bin/smoke_jax.sh # buildkit747 B

14

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

134.14 MB

15

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

199 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

7.53 GB

17COPY --chmod=0755 scripts/setup_jax_env.sh /usr/local/bin/setup_jax_env #9.73 KB

18ENV PYTHONUNBUFFERED=10 B

Command

 ARG RELEASE