Step for LARC 'local_lr' on 'p'
larc_step(p, local_lr, grad_avg = NULL, ...)
p
local_lr
local learning rate
grad_avg
gradient average
...
additional args to pass
None