jax

85150415 - Set the default of `jax_dump_ir_modes` to `stablehlo`. Fixes https://github.com/jax-ml/jax/issues/29819