jax

414449e1 - Merge pull request #26078 from gnecula:debug_info_jaxpr