jax.core.ClosedJaxpr

class jax.core.ClosedJaxpr(jaxpr, consts)

Parameters

• jaxpr (Jaxpr) –

• consts (Sequence) –

__init__(jaxpr, consts)

Initialize self. See help(type(self)) for accurate signature.

Parameters

• jaxpr (Jaxpr) –

• consts (Sequence) –

Methods

__init__(jaxpr, consts)	Initialize self.
map_jaxpr(f)

Attributes

eqns
in_avals
literals
out_avals