jax

c4e0db6f - [better_errors] Port the Pallas debug info mechanisms to the new JAX DebugInfo.

[better_errors] Port the Pallas debug info mechanisms to the new JAX DebugInfo. Now that we carry debug informatiion in Jaxpr we can remove the Pallas-specific tracking of the `func_src_info`, e.g., `NameAndSrcInfo`.