jax.core.ClosedJaxpr# class jax.core.ClosedJaxpr(jaxpr, consts)[source]# Parameters: jaxpr (Jaxpr) – consts (Sequence) – __init__(jaxpr, consts)[source]# Parameters: jaxpr (Jaxpr) – consts (Sequence) – Methods __init__(jaxpr, consts) param jaxpr: map_jaxpr(f) pretty_print(*[, source_info, print_shapes, ...]) replace(*[, jaxpr, consts]) Attributes consts effects eqns in_avals jaxpr literals out_avals