code: def __init__(
      self,
      lr: float,
      beta1: float,
      beta2: float,
      eps: float,
      weight_decay: float,
      use_adamw: bool,
  ) -> None:
      self.lr = lr
      self.beta1 = beta1
      self.beta2 = beta2
      self.eps = eps
      self.weight_decay = weight_decay
      self.use_adamw = use_adamw
  
imports: import math
  from abc import abstractmethod
  from typing import Type
  import pytest
  import torch
  from torch import Tensor
  from colossalai.utils import get_current_device, multi_tensor_applier
label_desc: Use image node_id 1 to create a new AdamKernel object  with example: obj = AdamKernel(lr, beta1, beta2, eps, weight_decay, use_adamw)