salt hashes?

Author: olupton

.github/container/Dockerfile.jax

@@ -48,7 +48,7 @@ RUN ARCH="$(dpkg --print-architecture)" && \
 # TODO: move this patch into the manifest
 ADD xla-arm64-neon.patch /opt
 RUN build-jax.sh \
-    --bazel-cache http://10.66.187.93:8080/cuda-12.6.2 \
+    --bazel-cache ${BAZEL_CACHE} \
     --build-path-jaxlib ${BUILD_PATH_JAXLIB} \
     --src-path-jax ${SRC_PATH_JAX} \
     --src-path-xla ${SRC_PATH_XLA} \

.github/container/build-jax.sh

@@ -196,6 +196,7 @@ fi
 if [[ "${BAZEL_CACHE}" == http://* ]] || \
    [[ "${BAZEL_CACHE}" == grpc://* ]]; then
     BUILD_PARAM="${BUILD_PARAM} --bazel_options=--remote_cache=${BAZEL_CACHE}"
+    BUILD_PARAM="${BUILD_PARAM} --bazel_options=--remote_default_platform_properties=jax_toolbox_toolchain=ubuntu22.04"
 elif [[ ! -z "${BAZEL_CACHE}" ]] ; then
     BUILD_PARAM="${BUILD_PARAM} --bazel_options=--disk_cache=${BAZEL_CACHE}"
 fi