jax

10ba2477 - Add a jaxpr_html IR dump mode that dumps a jaxpr to an HTML document.

Add a jaxpr_html IR dump mode that dumps a jaxpr to an HTML document. This allows us to visualize more information than is easily renderable in a static textual dump, in particular, the full source traceback. PiperOrigin-RevId: 910262639