Update benchmarks.yml

Author: juliagmt-google

.github/workflows/benchmarks.yml

@@ -34,21 +34,21 @@ jobs:
         env:
           JAXCI_CLONE_MAIN_XLA: 1
         run: ./ci/build_artifacts.sh "jaxlib"
-      - name: Build jax-cuda-plugin
-        env:
-          JAXCI_CLONE_MAIN_XLA: 1
-        run: ./ci/build_artifacts.sh "jax-cuda-plugin"
-      - name: Build jax-cuda-pjrt
-        env:
-          JAXCI_CLONE_MAIN_XLA: 1
-        run: ./ci/build_artifacts.sh "jax-cuda-pjrt"
+      # - name: Build jax-cuda-plugin
+      #   env:
+      #     JAXCI_CLONE_MAIN_XLA: 1
+      #   run: ./ci/build_artifacts.sh "jax-cuda-plugin"
+      # - name: Build jax-cuda-pjrt
+      #   env:
+      #     JAXCI_CLONE_MAIN_XLA: 1
+      #   run: ./ci/build_artifacts.sh "jax-cuda-pjrt"
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
           pip install pytest
           pip install absl-py
           # pip install -U jax
-          # pip install -U "jax[cuda12]"
+          pip install -U "jax[cuda12]"
           pip install google-benchmark
       - name: Run Multiprocess GPU Test
         run: |