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

    NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.