code: def _test_adamw(self, steps, opts, atol, rtol):
      self._test_optim(
          AdamW, torch.optim.AdamW, steps, opts, atol, rtol
      )
  
imports: import numpy
  import torch
  import unittest
  from tinygrad import Tensor, Device
  from tinygrad.nn.optim import Adam, SGD, AdamW
  from tinygrad.helpers import CI
label_desc: Use image node_id 4 for calling the TestOptim obj's underlying member method code with example usage: obj._test_adamw(steps, opts, atol, rtol) without return types