[ROCm] ci build and dockerfile changes#25148

copybara-service[bot] merged 1 commit intojax-ml:mainjax-ml/jax:mainfrom

ROCm:ci_install_sys_libs-upstreamROCm/jax:ci_install_sys_libs-upstreamCopy head branch name to clipboard