code: def larc_step(p, local_lr, grad_avg=None, **kwargs):
      "Step for LARC `local_lr` on `p`"
      p.data.add_(
          p.grad.data if grad_avg is None else grad_avg, alpha=-local_lr
      )
  
imports: from __future__ import annotations
  from .torch_basics import *
label_desc: Use image node_id 20 for calling a global function with example usage: larc_step(p, local_lr, grad_avg) without return types