NVIDIA
nemo-automodel
Container
NVIDIA
nemo-automodel

NVIDIA NeMo™ AutoModel accelerates LLM and VLM training and fine‑tuning with PyTorch DTensor‑native SPMD, day‑0 Hugging Face support, and optimized parallelism from single‑ to multi‑node scale.

  • LayerLabelCreated
    sha256:222c7135a27131c4da960f681611dc18d544d953778eb5b988e39917c0bd5295RUN
    INSTALL_TE=True TE_COMMIT=release_v2.11 DEEPEP_COMMIT=7febc6e25660af0f54d95dd781ecdcd62265ecca INSTALL_BITSANDBYTES=True BITSANDBYTES_COMMIT=0.49.2 INSTALL_UCCL_EP=False BASE_IMAGE=pytorch AUTOMODEL_INSTALL=all UV_SYNC_ARGS=--locked TARGETARCH=arm64 NVIDIA_BUILD_ID=305635189 NVIDIA_BUILD_REF=2c14eae97962250fe8f28c8f76cc3ae95ec110cf RC_DATE=26.04 echo "This distribution includes open source which is archived at the following URL: https://opensource.nvidia.com/oss/teams/nvidia/nemo-automodel/${RC_DATE}:linux-${TARGETARCH}/index.html" > NOTICES.txt &&
      echo "For further inquiries or assistance, contact us at oss-requests@nvidia.com" >> NOTICES.txt
    04/26/2026 8:08 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    TARGETARCH=arm64
    04/26/2026 8:08 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    RC_DATE=26.04
    04/26/2026 8:08 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
    com.nvidia.build.ref=2c14eae97962250fe8f28c8f76cc3ae95ec110cf
    04/26/2026 8:08 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_REF=2c14eae97962250fe8f28c8f76cc3ae95ec110cf
    04/26/2026 8:08 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
    com.nvidia.build.id=305635189
    04/26/2026 8:08 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_BUILD_ID=305635189
    04/26/2026 8:08 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_ID=305635189
    04/26/2026 8:08 AM UTC
    sha256:ab5b8c44d6011e3327d7ebbaa3f1344fe1bd2f1cdb9e951516fd227b13669b3cRUN
    INSTALL_TE=True TE_COMMIT=release_v2.11 DEEPEP_COMMIT=7febc6e25660af0f54d95dd781ecdcd62265ecca INSTALL_BITSANDBYTES=True BITSANDBYTES_COMMIT=0.49.2 INSTALL_UCCL_EP=False BASE_IMAGE=pytorch AUTOMODEL_INSTALL=all UV_SYNC_ARGS=--locked TARGETARCH=arm64 chmod +x /opt/venv/env.sh
    04/26/2026 8:08 AM UTC
    sha256:dee979fa3d16ec687d00620ca27bcb6acffddbd1ad407673e662aab1a10d2c41COPY
    <<EOF /opt/venv/env.sh
    04/26/2026 8:08 AM UTC
    ...