ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

oryx.core.interpreters.propagate.Equation

Hashable wrapper for jax_core.Jaxprs.

params

invars Dataclass field
outvars Dataclass field
primitive Dataclass field
params_flat Dataclass field
params_tree Dataclass field

Methods

from_jaxpr_eqn

View source

__eq__