[ROCm] ci build and dockerfile changes#161

Ruturaj4 wants to merge 1 commit intomainROCm/jax:mainfrom

ci_install_sys_libsROCm/jax:ci_install_sys_libsCopy head branch name to clipboard