buildx failed with: ERROR: failed to solve: process "/bin/sh -c build-jax.sh --bazel-cache ${BAZEL_CACHE} --src-path-jax ${SRC_PATH_JAX} --src-path-xla ${SRC_PATH_XLA} --sm all --xla-arm64-patch /opt/xla-arm64-neon.patch --other-xla-patches /opt/Mark-collective-called-computations-in-custom-shardi.patch --clean" did not complete successfully: exit code: 1

Node.js 16 actions are deprecated. Please update the following actions to use Node.js 20: webfactory/ssh-agent@v0.8.0, actions/checkout@v3, docker/login-action@v2, docker/setup-buildx-action@v2, docker/metadata-action@v4, docker/build-push-action@v4, actions/upload-artifact@v3. For more information see: https://github.blog/changelog/2023-09-22-github-actions-transitioning-from-node-16-to-node-20/.

Failed to download action 'https://api.github.com/repos/actions/checkout/tarball/f43a0e5ff2bd294095638e18286ca9a3d1956744'. Error: Response status code does not indicate success: 502 (Bad Gateway).