code: def momentum_step(p, lr, grad_avg, **kwargs):
      "Step for SGD with momentum with `lr`"
      p.data.add_(grad_avg, alpha=-lr)
  
imports: from __future__ import annotations
  from .torch_basics import *
label_desc: Use image node_id 7 for calling a global function with example usage: momentum_step(p, lr, grad_avg) without return types