code: def MuAdamW(params, **kwargs):
      """AdamW with μP scaling.
  
      Note for this to work properly, your model needs to have its base shapes set
      already using `mup.set_base_shapes`.
      """
      return MuAdam(params, impl=AdamW, **kwargs)
  
imports: from collections import defaultdict
  from torch.optim import SGD, Adam, AdamW
label_desc: Use image node_id 3 for calling a global function with example usage: MuAdamW(params) and returns: MuAdam