ci: Install CI dependencies from uv lock

.coveragerc

@@ -27,10 +27,14 @@ omit =
     /workspace/config-3.12.py
     /workspace/config-3.py
     /workspace/config.py
+    /workspace/_remote_module_non_scriptable*
+
+[report]
+omit =
+    /workspace/_remote_module_non_scriptable*
 
 [paths]
 source =
     nemo/
     /home/runner/work/NeMo/NeMo/nemo
     /workspace/nemo
-

.github/actions/test-template/action.yml

@@ -153,7 +153,6 @@ runs:
           set -e
 
           docker exec -t nemo_container_${{ github.run_id }}_${{ inputs.runner }} bash -c '\
-            cp -r /opt/Megatron-LM/ /workspace/ && \
             bash tests/${{ inputs.test_dir }}/${{ inputs.script }}.sh && \
             echo "Finished successfully." || echo "Did not finish."'
         ) 2>&1 | tee $DIR/err.log

.github/workflows/_build_container.yml

@@ -38,26 +38,6 @@ jobs:
       - name: Checkout repository
         uses: actions/checkout@v6
 
-      - name: Parse manifest.json
-        id: manifest
-        run: |
-          BUILD_ARGS=$(cat << EOF
-          BASE_IMAGE=$(cat requirements/manifest.json | jq -r '."ngc-pytorch"')
-          TRTLLM_REPO=$(cat requirements/manifest.json | jq -r '."vcs-dependencies"."trt-llm".repo')
-          TRTLLM_TAG=$(cat requirements/manifest.json | jq -r '."vcs-dependencies"."trt-llm".ref')
-          MLM_REPO=$(cat requirements/manifest.json | jq -r '."vcs-dependencies"."megatron-lm".repo')
-          MLM_TAG=$(cat requirements/manifest.json | jq -r '."vcs-dependencies"."megatron-lm".ref')
-          TE_REPO=$(cat requirements/manifest.json | jq -r '."vcs-dependencies".transformer_engine.repo')
-          TE_TAG=$(cat requirements/manifest.json | jq -r '."vcs-dependencies".transformer_engine.ref')
-          APEX_REPO=$(cat requirements/manifest.json | jq -r '."vcs-dependencies".apex.repo')
-          APEX_TAG=$(cat requirements/manifest.json | jq -r '."vcs-dependencies".apex.ref')
-          EOF
-          )
-
-          echo "BUILD_ARGS<<EOF" >> $GITHUB_OUTPUT
-          echo "$BUILD_ARGS" >> $GITHUB_OUTPUT
-          echo "EOF" >> $GITHUB_OUTPUT
-
       - name: Get last merged PR
         id: cache_from
         env:
@@ -121,7 +101,6 @@ jobs:
             NEMO_TAG=${{ github.sha }}
             NEMO_REPO=https://github.com/NVIDIA/NeMo
             PR_NUMBER=${{ github.event.pull_request.number || 0 }}
-            ${{ needs.pre-flight.outputs.build_args }}
           cache-from: |
             type=registry,ref=${{ inputs.registry }}/nemo-speech:${{ inputs.image-name }}-buildcache-main,mode=max
             type=registry,ref=${{ inputs.registry }}/nemo-speech:${{ inputs.image-name }}-buildcache-${{ github.event.pull_request.number || 0 }},mode=max

CLAUDE.md

@@ -49,9 +49,9 @@ Markers: `unit`, `integration`, `system`, `pleasefixme` (broken — skip), `skip
 Sphinx-based docs live in `docs/source/`. Build with:
 
 ```bash
-pip install -r requirements/requirements_docs.txt   # one-time setup
-make -C docs clean html                              # full rebuild
-make -C docs html                                    # incremental rebuild
+uv sync --group docs                                 # one-time setup
+uv run make -C docs clean html                       # full rebuild
+uv run make -C docs html                             # incremental rebuild
 ```
 
 Output goes to `docs/build/html/`. Open `docs/build/html/index.html` to preview locally.

docker/Dockerfile.ci

@@ -14,71 +14,53 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:24.07-py3
+ARG BASE_IMAGE=nvcr.io/nvidia/cuda-dl-base:26.04-cuda13.2-devel-ubuntu24.04
+FROM ${BASE_IMAGE} AS base-image
+ARG UV_VERSION=0.11.14
 
-FROM ${BASE_IMAGE} as base-image
-ENV PIP_CONSTRAINT="" 
+ENV DEBIAN_FRONTEND=noninteractive
 ENV TRANSFORMERS_OFFLINE=0
 ENV HYDRA_FULL_ERROR=1
 ENV PYTHONUNBUFFERED=1
+ENV UV_PYTHON=3.13
+ENV UV_PROJECT_ENVIRONMENT=/opt/venv
+ENV UV_LINK_MODE=copy
+ENV UV_NO_CACHE=1
+ENV UV_NO_PROGRESS=1
+ENV VIRTUAL_ENV=/opt/venv
+ENV PATH="/root/.local/bin:${VIRTUAL_ENV}/bin:${PATH}"
 
-# APT packages
 RUN <<"EOF" bash -ex
 apt-get update
-apt-get install -y bc libsox-fmt-all
+apt-get install -y \
+    bc \
+    curl \
+    git \
+    libsndfile1 \
+    libsox-fmt-all \
+    sox
 apt-get clean
 EOF
 
-FROM base-image as te-wheel
-WORKDIR /tmp/NeMo
-ARG TE_REPO
-ARG TE_TAG
-RUN --mount=type=bind,source=docker/common/install_dep.sh,target=/tmp/NeMo/install_dep.sh \
-  --mount=type=bind,source=external/patches,target=/tmp/NeMo/external/patches <<"EOF" bash -ex
-
-  bash /tmp/NeMo/install_dep.sh --library te --mode build
-  ls -al /tmp/Megatron-LM || true
-EOF
-
-FROM base-image as mcore-wheel
-WORKDIR /tmp/NeMo
-ARG MLM_REPO
-ARG MLM_TAG
-RUN --mount=type=bind,source=docker/common/install_dep.sh,target=/tmp/NeMo/install_dep.sh <<"EOF" bash -ex
-
-  bash /tmp/NeMo/install_dep.sh --library mcore --mode build
-  ls -al /tmp/Megatron-LM || true
+RUN <<"EOF" bash -ex
+curl -LsSf "https://astral.sh/uv/${UV_VERSION}/install.sh" | sh
+uv --version
 EOF
 
-FROM base-image 
-WORKDIR /tmp/NeMo
-ENV INSTALL_DIR="/opt"
-RUN \  
-  --mount=type=bind,from=te-wheel,source=/opt/wheels/te,target=/opt/wheels/te \
-  --mount=type=bind,from=mcore-wheel,source=/opt/wheels/mcore,target=/opt/wheels/mcore \
-  --mount=type=bind,source=requirements,target=/tmp/NeMo/requirements \
-  --mount=type=bind,source=tools/ctc_segmentation/requirements.txt,target=/tmp/NeMo/tools/ctc_segmentation/requirements.txt \
-  --mount=type=bind,source=docker/common/install_dep.sh,target=/tmp/NeMo/install_dep.sh \
-  --mount=type=bind,source=setup.py,target=/tmp/NeMo/setup.py \
-  --mount=type=bind,source=external/patches,target=/tmp/NeMo/external/patches \
-  --mount=type=bind,source=README.md,target=/tmp/NeMo/README.md \
-  --mount=type=bind,source=nemo/package_info.py,target=/tmp/NeMo/nemo/package_info.py \
-  --mount=type=bind,source=nemo/__init__.py,target=/tmp/NeMo/nemo/__init__.py <<"EOF" bash -ex
-
-  bash /tmp/NeMo/install_dep.sh --library te --mode install
-  bash /tmp/NeMo/install_dep.sh --library mcore --mode install
-  bash /tmp/NeMo/install_dep.sh --library extra --mode install
-  pip install --no-cache-dir ".[all,cu12]"
-  rm -rf $NEMO_DIR || true
+WORKDIR /workspace
+COPY pyproject.toml uv.lock /workspace/
+COPY nemo/__init__.py nemo/package_info.py /workspace/nemo/
+RUN <<"EOF" bash -ex
+uv sync --link-mode copy --locked --extra all --extra cu13 --group test
 EOF
 
-
-WORKDIR /workspace
-ENV PYTHONPATH="${PYTHONPATH}:/workspace/Megatron-LM"
 ENV NEMO_HOME="/home/TestData/nemo_home"
 
 ARG IMAGE_LABEL
 LABEL "nemo.library"=${IMAGE_LABEL}
 
 ARG PR_NUMBER
 LABEL "nemo.pr_number"=${PR_NUMBER}
+
+ARG NVIDIA_BUILD_ID
+LABEL "NVIDIA_BUILD_ID"=${NVIDIA_BUILD_ID}