code: def __init__(
      self, init_scale: float = 2**32, growth_interval: int = 1000
  ) -> None:
      super().__init__(precision=16)
  
      self.scaler = torch.cuda.amp.GradScaler(
          init_scale=init_scale, growth_interval=growth_interval
      )
  
imports: import torch
  from pytorch_lightning.plugins.precision import MixedPrecisionPlugin
label_desc: Use image node_id 1 to create a new NeMoMixedPrecisionPlugin object from inherited base classes: MixedPrecisionPlugin with example: obj = NeMoMixedPrecisionPlugin(init_scale, growth_interval)