oryx.experimental.optimizers.optix.clip_by_global_norm

Returns a function that clips updates to a provided max norm.