Save the date! Google I/O returns May 18-20 Register now

oryx.experimental.optimizers.optix.scale_by_rms

Returns a function that scales updates by the RMS of the updates.