Fix bazel options

Author: Steboss

.github/container/Dockerfile.jax

@@ -58,13 +58,13 @@ RUN --mount=type=bind,from=bazel_disk,target=/cache/bazel-disk,rw \
     build-jax.sh \
     --bazel-cache ${BAZEL_CACHE} \
     --build-path-jaxlib ${BUILD_PATH_JAXLIB} \
-    --bazel-options "--jobs=28 --local_ram_resources=HOST_RAM*.8" \
     --extra-targets "${EXTRA_BAZEL_TARGETS}" \
     --extra-target-dest /builder/extra-targets \
     --no-install \
     --src-path-jax ${SRC_PATH_JAX} \
     --src-path-xla ${SRC_PATH_XLA} \
     --clean \
+    --build-param=--bazel_options="--jobs=28 --local_ram_resources=HOST_RAM*.8" \
     ${EXTRA_BUILD_JAX_ARGS}
 
 ## Transformer engine: check out source and build wheel