Support wheels as build deps#638

alekstheod wants to merge 4 commits intorocm-jaxlib-v0.8.2ROCm/jax:rocm-jaxlib-v0.8.2from

support_wheels_as_build_depsROCm/jax:support_wheels_as_build_depsCopy head branch name to clipboard