oryx.experimental.optimizers.optix.scale_by_schedule

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