class_docstr: Wrapper which takes in a KeyedOptimizer and is a KeyedOptimizer
  
  Subclass for Optimizers like GradientClippingOptimizer and WarmupOptimizer
class_docstr_tok: ["Wrapper","which","takes","in","a","KeyedOptimizer","and","is","a","KeyedOptimizer","Subclass","for","Optimizers","like","GradientClippingOptimizer","and","WarmupOptimizer"]
code: def state_dict(self) -> Dict[str, Any]:
      return self._optimizer.state_dict()
  
imports: import json
  from copy import deepcopy
  from typing import Any, Callable, Collection, Dict, List, Mapping, Optional, OrderedDict, Set, Tuple, Union
  import torch
  from torch import optim
  from torchrec.distributed.types import ShardedTensor
label_desc: Use image node_id 6 for calling the OptimizerWrapper obj's underlying member method code with example usage: obj.state_dict() and returns: self