def METHOD_NAME(self, event: Event, state: State, logger: Logger) -> Optional[int]:
    del event, logger  # unused
    apply_low_precision_layernorm(model=state.model, optimizers=state.optimizers, precision=state._precision)