Rework plugin/PJRT wheel build to use Bazel Build#328

lucbruni-amd wants to merge 1 commit intomasterROCm/rocm-jax:masterfrom

lb/bazel-build-wheelhouseROCm/rocm-jax:lb/bazel-build-wheelhouseCopy head branch name to clipboard