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

oryx.experimental.optimizers.optix.scale_by_schedule

Returns a function that scales updates according to an input schedule.