jax

c0c4c019 - [ROCm] Support lowering through PJRT_Triton_Extension

[ROCm] Support lowering through PJRT_Triton_Extension This change adds support for lowering Pallas Triton calls to HSACO through PJRT_Triton_Extension for ROCm.