class_docstr: A wrapper for pytorch.optim.Optimizer that forwards all methods to the wrapped optimizer
class_docstr_tok: ["A","wrapper","for","pytorch.optim.Optimizer","that","forwards","all","methods","to","the","wrapped","optimizer"]
code: def zero_grad(self, *args, **kwargs):
      return self.optim.zero_grad(*args, **kwargs)
  
file_path: hivemind/hivemind/moe/server/layers/optim.py
imports: import torch
label_desc: Use image node_id 10 for calling the OptimizerWrapper obj's underlying member method code with example usage: obj.zero_grad() and returns: self