Add build targets for jax-rocm-plugin
and jax-rocm-pjrt
wheels.
#5244
Loading
jax-rocm-plugin
and jax-rocm-pjrt
wheels.
#5244