oryx.experimental.matching.jax_rewrite.evaluate

Evaluates expressions into JAX values.

oryx.experimental.matching.jax_rewrite.evaluate(
    obj,
    env: oryx.experimental.matching.jax_rewrite.Bindings
) -> Any

evaluate is a single-dispatch function whose default implementation is to raise a NotImplementedError. To register evaluation functions for particular types, use evaluate.register.

Args
obj	an object to evaluate.
env	a dictionary mapping string names to JAX values

Returns
The JAX value that is the result of evaluating the object.