class_docstr: Megatron specific extensions of torch Module with support
  for pipelining.
docstr: Use this function to override the state dict for
  saving checkpoints.
docstr_tok: ["Use","this","function","to","override","the","state","dict","for","saving","checkpoints","."]
code: def state_dict_for_save_checkpoint(
      self, destination=None, prefix="", keep_vars=False
  ):
      """Use this function to override the state dict for
      saving checkpoints."""
      return self.state_dict(destination, prefix, keep_vars)
  
imports: import torch
  from torch.autograd import Variable
  from torch.nn.parameter import Parameter
  from nemo.collections.nlp.modules.common.megatron.utils import ApexGuardDefaults
  from nemo.utils import logging
label_desc: Use image node_id 10 for calling the MegatronModule obj's underlying member method code with example usage: obj.state_dict_for_save_checkpoint(destination, prefix, keep_vars) and returns: self