jax.core.Jaxpr

class jax.core.Jaxpr(constvars, invars, outvars, eqns)[source]
__init__(constvars, invars, outvars, eqns)[source]
Params:
constvars: list of variables introduced for constants (either literals
in the Python program, or the result of constant folding during the generation of the Jaxpr). Array constants are replaced with such variables while scalar constants are kept inline.
invars: list of input variables. Together, constvars and invars are
the inputs to the Jaxpr.

outvars: list of output variables. eqns: list of equations.

Methods

__init__(constvars, invars, outvars, eqns) Params: