oryx.experimental.matching.rules.tree_map

Shallow maps a function over the children of an expression.

Main aliases

oryx.experimental.matching.jax_rewrite.rules.tree_map