oryx.core.trace_util.trees

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

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