oryx.core.trace_util.trees

Returns a function that determines input and output pytrees from inputs.

Main aliases

oryx.experimental.matching.jax_rewrite.trace_util.trees