oryx.experimental.matching.jax_rewrite.evaluate

Stay organized with collections Save and categorize content based on your preferences.

Evaluates expressions into JAX values.

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

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

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