code: def sgd_step(p, lr, **kwargs):
      p.data.add_(p.grad.data, alpha=-lr)
  
imports: from __future__ import annotations
  from .torch_basics import *
label_desc: Use image node_id 2 for calling a global function with example usage: sgd_step(p, lr) without return types