jax

5479321e - Merge pull request #34832 from ROCm:add-rocm-device-placement-compat-test