Sign inSign up

rocm/jax:rocm10.0-jax0.10.1-py3.13

Manifest digest

sha256:8597e93f6d6b730f5da0fc7ce61b52fca8ebc55b3b75c2d3ccfb801cfd938a27

OS/ARCH

linux/amd64

Compressed size

7.78 GB

Last pushed

26 days by rocm

Type

Image

Manifest digest

sha256:8597e93f6d6b730f5da0fc7ce61b52fca8ebc55b3b75c2d3ccfb801cfd938a27

Image Layers

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.version=24.040 B
4ADD file ... in / 28.37 MB
5CMD ["/bin/bash"]0 B
6ARG PYTHON_VERSION=3.130 B
7ARG ROCM_VERSION=10.0.00 B
8ARG JAX_VERSION=0.10.10 B
9ARG INDEX_URL=https://stable.repo.amd.com/rocm/whl-next0 B
10ENV DEBIAN_FRONTEND=noninteractive0 B
11LABEL image.title=JAX ROCm Nightly image.vendor=Unified0 B
12COPY scripts/nightly/install_python.sh /tmp/install_python.sh # buildkit1.05 KB
13COPY jax-src /workspace/jax # buildkit35.5 MB
14RUN |4 PYTHON_VERSION=3.13 ROCM_VERSION=10.0.0 JAX_VERSION=0.10.1217.21 MB
15RUN |4 PYTHON_VERSION=3.13 ROCM_VERSION=10.0.0 JAX_VERSION=0.10.153.33 MB
16ENV PATH=/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin VIRTUAL_ENV=/opt/venv0 B
17RUN |4 PYTHON_VERSION=3.13 ROCM_VERSION=10.0.0 JAX_VERSION=0.10.198.86 MB
18RUN |4 PYTHON_VERSION=3.13 ROCM_VERSION=10.0.0 JAX_VERSION=0.10.17.13 GB
19ENV ROCM_PATH=/opt/venv/lib/python3.13/site-packages/_rocm_sdk_core ROCM_HOME=/opt/venv/lib/python3.13/site-packages/_rocm_sdk_core HIP_PATH=/opt/venv/lib/python3.13/site-packages/_rocm_sdk_core ROCM_VERSION=10.0.00 B
20RUN |4 PYTHON_VERSION=3.13 ROCM_VERSION=10.0.0 JAX_VERSION=0.10.1197 B
21RUN |4 PYTHON_VERSION=3.13 ROCM_VERSION=10.0.0 JAX_VERSION=0.10.17.49 KB
22RUN |4 PYTHON_VERSION=3.13 ROCM_VERSION=10.0.0 JAX_VERSION=0.10.1236.97 MB
23RUN |4 PYTHON_VERSION=3.13 ROCM_VERSION=10.0.0 JAX_VERSION=0.10.11.93 KB
24WORKDIR /0 B
25ENTRYPOINT ["/usr/local/bin/ufb-entrypoint.sh"]0 B
26CMD ["/bin/bash"]0 B

Command

 ARG RELEASE