code: def __init__(self, weight, bias=None, mp_group=None):
      super(LinearAllreduce, self).__init__()
      self.weight = weight
      self.bias = bias
      self.mp_group = mp_group
  
imports: import torch
  from deepspeed import comm
  from torch import nn
  from torch.nn import functional
  from torch.nn.parameter import Parameter
  from deepspeed.accelerator import get_accelerator
  from deepspeed.module_inject.tp_shard import get_shard_size, get_shard_size_list
label_desc: Use image node_id 1 to create a new LinearAllreduce object from inherited base classes: nn with example: obj = LinearAllreduce(weight, bias, mp_group)