code: def __init__(self, data: Tensor, group: "ProcessGroup"):
      self.data = data
      self.group = group
      self.offset = 0
      self.callbacks: List[Callable] = []
      self.output_shard = torch.zeros_like(data[0])
  
file_path: fairscale/fairscale/internal/reduce_scatter_bucketer.py
imports: import functools
  import os
  from typing import TYPE_CHECKING, Callable, Dict, List, Optional, Tuple
  import torch
  from torch import Tensor
  import torch.distributed
label_desc: Use image node_id 1 to create a new Bucket object  with example: obj = Bucket(data, group)