jax.core.ClosedJaxpr

jax.core.ClosedJaxpr#

class jax.core.ClosedJaxpr(jaxpr, consts)[source]#
Parameters:
__init__(jaxpr, consts)[source]#
Parameters:

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