oryx.experimental.optimizers.optix.scale_by_rms

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