class_docstr: Precision plugin for DeepSpeed integration.
  
  Args:
      precision: Full precision (32-true), half precision (16-true, bf16-true) or
          mixed precision (16-mixed, bf16-mixed).
  
  Raises:
      ValueError:
          If unsupported ``precision`` is provided.
class_docstr_tok: ["Precision","plugin","for","DeepSpeed","integration",".","Args",":","precision",":","Full","precision","(","32-true",")",",","half","precision","(","16-true",",","bf16-true",")","or","mixed","precision","(","16-mixed",",","bf16-mixed",")",".","Raises",":","ValueError",":","If","unsupported","``","precision","``","is","provided","."]
code: def module_init_context(self) -> ContextManager:
      return self.tensor_init_context()
  
imports: from contextlib import nullcontext
  from typing import TYPE_CHECKING, Any, ContextManager, Literal
  import torch
  from lightning_utilities.core.apply_func import apply_to_collection
  from torch import Tensor
  from torch.nn import Module
  from typing_extensions import get_args, override
  from lightning.fabric.plugins.precision.precision import Precision
  from lightning.fabric.plugins.precision.utils import _convert_fp_tensor, _DtypeContextManager
  from lightning.fabric.utilities.types import Steppable
label_desc: Use image node_id 4 for calling the DeepSpeedPrecision obj's underlying member method code with example usage: obj.module_init_context() and returns: self