Internal APIs

core

Jaxpr(constvars, invars, outvars, eqns)
TypedJaxpr(jaxpr, literals, in_avals, out_avals)