jax

4e442000 - Concatenate caller and callee tracebacks in jaxpr equation profiles.

Concatenate caller and callee tracebacks in jaxpr equation profiles. We changed jaxprs so we only keep tracebacks up to the enclosing jit on each equation. However, when forming an equation profile, we need the full traceback. PiperOrigin-RevId: 899188530