jax

000b92f5 - [better_errors] Continue adding debug info to Jaxprs (step 5)

[better_errors] Continue adding debug info to Jaxprs (step 5) This follows in a series, starting with #26078 and #26313, adding debug_info to more calls to lu.wrap_init. These changes ensure that all the lu.wrap_init and Jaxpr are called with debug_info in the api_test.py:CustomTransposeTest, api_test.py:CustomVmapTest and api_test.py:RematTest.