From 5f342e85914648bfa80d49f872e38f3c901abc46 Mon Sep 17 00:00:00 2001 From: Vladislav Date: Thu, 8 Aug 2024 23:52:53 -0700 Subject: [PATCH] Apply suggestions from Brain and googlers: minnor fixes --- .github/container/build-jax.sh | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/container/build-jax.sh b/.github/container/build-jax.sh index ea8209139..92198a055 100755 --- 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