jax

7ac80f26 - [debug_info] Minor fix following https://github.com/jax-ml/jax/pull/30012