oryx.experimental.matching.jax_rewrite.make_bound_expression

Returns a function that traces a function to produce an expression.