oryx.core.trace_util.stage

Returns a function that stages a function to a ClosedJaxper.

Main aliases

oryx.experimental.matching.jax_rewrite.trace_util.stage