[Issue]: Jax's 3rd party Bazel Download intermittent failure.

[JeniferC99]

Problem Description

Job: Release portable Linux JAX Wheels (gfx120X-all, prerelease, 7.12.0rc0) · ROCm/rockrel@ec608b3

Error log:
WARNING: Download from https://github.com/ROCm/jax/archive/21d5e0bf8f07fb7e455f627284e490e6e3068176.tar.gz failed: class java.io.IOException GET returned 504 Gateway Time-out
INFO: Repository jax instantiated at:
/jax_rocm_plugin/WORKSPACE:53:17: in
/jax_rocm_plugin/third_party/jax/workspace.bzl:8:21: in repo
/jax_rocm_plugin/third_party/repo.bzl:204:22: in amd_http_archive
Repository rule _amd_http_archive defined at:
/jax_rocm_plugin/third_party/repo.bzl:190:36: in
ERROR: /jax_rocm_plugin/third_party/repo.bzl:176:29: An error occurred during the fetch of repository 'jax':
Traceback (most recent call last):
  File "/jax_rocm_plugin/third_party/repo.bzl", line 176, column 29, in _amd_http_archive_impl
    ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error downloading [https://github.com/ROCm/jax/archive/21d5e0bf8f07fb7e455f627284e490e6e3068176.tar.gz] to /root/.cache/bazel/_bazel_root/35c03de1b96cb539bbf04926e943e4d4/external/jax/temp11400760131813579037/21d5e0bf8f07fb7e455f627284e490e6e3068176.tar.gz: GET returned 504 Gateway Time-out
ERROR: Error computing the main repository mapping: at /root/.cache/bazel/_bazel_root/35c03de1b96cb539bbf04926e943e4d4/external/xla/third_party/py/python_init_toolchains.bzl:4:5: Encountered error while reading extension file 'py_version.bzl': no such package '@@python_version_repo//': no such package '@@jax//build': java.io.IOException: Error downloading [https://github.com/ROCm/jax/archive/21d5e0bf8f07fb7e455f627284e490e6e3068176.tar.gz] to /root/.cache/bazel/_bazel_root/35c03de1b96cb539bbf04926e943e4d4/external/jax/temp11400760131813579037/21d5e0bf8f07fb7e455f627284e490e6e3068176.tar.gz: GET returned 504 Gateway Time-out

Hitting intermitent gateway time out, 3rd party installation should be removed from build step.

Operating System

Ubuntu

CPU

n/a

GPU

n/a

ROCm Version

ROCm 7.12

ROCm Component

No response

Steps to Reproduce

No response

(Optional for Linux users) Output of /opt/rocm/bin/rocminfo --support

No response

Additional Information

No response