jax.core.ClosedJaxpr ==================== .. currentmodule:: jax.core .. autoclass:: ClosedJaxpr .. automethod:: __init__ .. rubric:: Methods .. autosummary:: ~ClosedJaxpr.__init__ ~ClosedJaxpr.map_jaxpr ~ClosedJaxpr.pretty_print ~ClosedJaxpr.replace .. rubric:: Attributes .. autosummary:: ~ClosedJaxpr.effects ~ClosedJaxpr.eqns ~ClosedJaxpr.in_avals ~ClosedJaxpr.literals ~ClosedJaxpr.out_avals ~ClosedJaxpr.jaxpr ~ClosedJaxpr.consts