diff --git a/3rdparty/NeMo b/3rdparty/NeMo
index abd4bf7e6..0cd990d97 160000
--- a/3rdparty/NeMo
+++ b/3rdparty/NeMo
@@ -1 +1 @@
-Subproject commit abd4bf7e6961856c04404452b50e7574dd7629e4
+Subproject commit 0cd990d97f01ab946fc53108ed43c7be7140a0d8
diff --git a/Dockerfile b/Dockerfile
index 1452a6f4b..7a760a4dd 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -105,18 +105,6 @@ COPY --from=rust-env /usr/local/rustup /usr/local/rustup
 ENV PATH="/usr/local/cargo/bin:/usr/local/rustup/bin:${PATH}"
 ENV RUSTUP_HOME="/usr/local/rustup"
 
-RUN --mount=type=cache,target=/root/.cache <<EOF
-set -eo pipefail
-uv pip install maturin --no-build-isolation
-
-pip install --use-deprecated=legacy-resolver  --no-build-isolation \
-  tensorstore==0.1.45
-sed -i 's/^Version: 0\.0\.0$/Version: 0.1.45/' \
-  /usr/local/lib/python3.12/dist-packages/tensorstore-0.0.0.dist-info/METADATA
-mv /usr/local/lib/python3.12/dist-packages/tensorstore-0.0.0.dist-info \
-/usr/local/lib/python3.12/dist-packages/tensorstore-0.1.45.dist-info
-EOF
-
 WORKDIR /workspace/bionemo2
 
 # Install 3rd-party deps and bionemo submodules.
@@ -133,6 +121,8 @@ RUN --mount=type=bind,source=./.git,target=./.git \
     --mount=type=cache,target=/root/.cache <<EOF
 set -eo pipefail
 
+uv pip install maturin --no-build-isolation
+
 uv pip install --no-build-isolation \
   ./3rdparty/* \
   ./sub-packages/bionemo-* \