don't install te in model requirements, use uv in dockerfiles

Signed-off-by: Peter St. John <pstjohn@nvidia.com>

Author: pstjohn

.github/workflows/framework-unit-tests.yml

@@ -59,7 +59,6 @@ jobs:
       - uses: actions/setup-python@v5
         with:
           python-version: "3.13"
-          cache: "pip"
       - name: Setup UV
         uses: astral-sh/setup-uv@v6
         with:

models/amplify/Dockerfile

@@ -4,7 +4,8 @@ FROM nvcr.io/nvidia/pytorch:25.01-py3
 RUN MAX_JOBS=4 pip --disable-pip-version-check --no-cache-dir install -v git+https://github.com/facebookresearch/xformers.git@v0.0.29.post1#egg=xformers
 RUN PIP_CONSTRAINT= NVTE_FRAMEWORK=pytorch MAX_JOBS=4 pip --disable-pip-version-check --no-cache-dir install -v git+https://github.com/nvidia/TransformerEngine.git@v2.4
 
+COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/
 WORKDIR /workspace/bionemo
 COPY . .
-RUN --mount=type=cache,target=/root/.cache/pip \
-    PIP_CONSTRAINT= pip install .
+RUN --mount=type=cache,target=/root/.cache/uv \
+    uv pip install --system --break-system-packages -e .

models/amplify/pyproject.toml

@@ -20,7 +20,7 @@ dependencies = [
     "omegaconf",
     "pytest",
     "torch",
-    "transformer-engine",
+    # "transformer_engine[pytorch]",
     "transformers",
     "xformers",
 ]

models/esm2/Dockerfile

@@ -1,5 +1,6 @@
 FROM nvcr.io/nvidia/pytorch:25.06-py3
+COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/
 WORKDIR /workspace/bionemo
 COPY . .
-RUN --mount=type=cache,target=/root/.cache/pip \
-    PIP_CONSTRAINT= pip install .
+RUN --mount=type=cache,target=/root/.cache/uv \
+    uv pip install --system --break-system-packages -e .

models/esm2/pyproject.toml

@@ -18,7 +18,7 @@ dependencies = [
     "omegaconf",
     "pytest",
     "torch",
-    "transformer-engine",
+    # "transformer_engine[pytorch]",
     "transformers",
 ]