jax.core.TypedJaxpr

class jax.core.TypedJaxpr(jaxpr, literals, in_avals, out_avals)[source]
__init__(jaxpr, literals, in_avals, out_avals)[source]

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

Methods

__init__(jaxpr, literals, in_avals, out_avals) Initialize self.