Apply suggestions from Brain and googlers: minnor fixes

NVIDIA · Aug 9, 2024 · 5f342e8 · 5f342e8
1 parent 5601263
commit 5f342e8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/.github/container/build-jax.sh b/.github/container/build-jax.sh
@@ -304,9 +304,9 @@ fi
 
 # after installation (example)
 #  jax                     0.4.32.dev20240808+9c2caedab /opt/jax
-#  jax-cuda12-pjrt         0.4.32.dev20240808           /opt/jaxlibs/jax_gpu_pjrt
-#  jax-cuda12-plugin       0.4.32.dev20240808           /opt/jaxlibs/jax_gpu_plugin
-#  jaxlib                  0.4.32.dev20240808           /opt/jaxlibs/jaxlib
+#  jax-cuda12-pjrt         0.4.32.dev20240808           /opt/jax/dist/jax_gpu_pjrt
+#  jax-cuda12-plugin       0.4.32.dev20240808           /opt/jax/dist/jax_gpu_plugin
+#  jaxlib                  0.4.32.dev20240808           /opt/jax/dist/jaxlib
 pip list | grep jax
 
 ## Cleanup