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

Manifest digest

sha256:0369efd4d1da6f4e361b99020f60f3149dd0e8c23a57d5ff1197b99a21f86b72

OS/ARCH

linux/amd64

Compressed size

7.77 GB

Last pushed

about 2 months by rocm

Type

Image

Manifest digest

sha256:0369efd4d1da6f4e361b99020f60f3149dd0e8c23a57d5ff1197b99a21f86b72

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.11.00 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 # buildkit39.4 MB

14RUN |4 PYTHON_VERSION=3.13 ROCM_VERSION=10.0.0 JAX_VERSION=0.11.0217.21 MB

15RUN |4 PYTHON_VERSION=3.13 ROCM_VERSION=10.0.0 JAX_VERSION=0.11.053.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.11.098.85 MB

18RUN |4 PYTHON_VERSION=3.13 ROCM_VERSION=10.0.0 JAX_VERSION=0.11.07.13 GB

19

ENV 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.0

0 B

20RUN |4 PYTHON_VERSION=3.13 ROCM_VERSION=10.0.0 JAX_VERSION=0.11.0197 B

21RUN |4 PYTHON_VERSION=3.13 ROCM_VERSION=10.0.0 JAX_VERSION=0.11.07.48 KB

22RUN |4 PYTHON_VERSION=3.13 ROCM_VERSION=10.0.0 JAX_VERSION=0.11.0223.64 MB

23RUN |4 PYTHON_VERSION=3.13 ROCM_VERSION=10.0.0 JAX_VERSION=0.11.01.93 KB

24WORKDIR /0 B

25ENTRYPOINT ["/usr/local/bin/ufb-entrypoint.sh"]0 B

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

Command

 ARG RELEASE