jax

aae3e281 - [ROCm] Bump XLA pin to ROCm/xla rocm-jaxlib-v0.9.2 HEAD (rocprofiler-sdk v3 + roctracer v1 build fix) (#766)