diff --git a/Dockerfile b/Dockerfile
index 7e94cb64a..e4279c9f7 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -146,8 +146,7 @@ RUN apt update -y && apt-get install -y google-perftools glibc-tools
 ENV PIP_FIND_LINKS=https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 RUN pip install .[core,gpu]
 COPY . .
-RUN pip install -U "jax[gpu]==0.4.38" "jax==0.4.38" "jaxlib==0.4.38" \
-    -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
+RUN pip install -U "jax[gpu]==0.4.38" "jax==0.4.38" "jaxlib==0.4.38"
 COPY . .