Internal APIs

core

Jaxpr(constvars, invars, outvars, eqns)

param constvars

ClosedJaxpr(jaxpr, consts)

param jaxpr