Google I/O returns May 18-20! Reserve space and build your schedule Register now

oryx.experimental.optimizers.optix.scale_by_schedule

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