ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

oryx.core.trace_util.stage

Returns a function that stages a function to a ClosedJaxper.