code: def rms_prop_step(p, lr, sqr_avg, eps, grad_avg=None, **kwargs):
      "Step for RMSProp with momentum with `lr`"
      denom = sqr_avg.sqrt().add_(eps)
      p.data.addcdiv_(
          (grad_avg if grad_avg is not None else p.grad),
          denom,
          value=-lr,
      )
  
imports: from __future__ import annotations
  from .torch_basics import *
label_desc: Use image node_id 9 for calling a global function with example usage: rms_prop_step(p, lr, sqr_avg, eps, grad_avg) without return types