code: def __init__(
      self,
      lr: float,
      beta1: float,
      beta2: float,
      eps: float,
      weight_decay: float,
      use_adamw: bool,
  ) -> None:
      super().__init__(lr, beta1, beta2, eps, weight_decay, use_adamw)
      from colossalai.kernel.op_builder import CPUAdamBuilder
  
      cpu_optim = CPUAdamBuilder().load()
  
      self.cpu_adam_op = cpu_optim.CPUAdamOptimizer(
          lr, beta1, beta2, eps, weight_decay, 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 CPUAdamKernel object from inherited base classes: AdamKernel with example: obj = CPUAdamKernel(lr, beta1, beta2, eps, weight_decay, use_adamw)