code: def __init__(self):
      super().__init__()
      self.fc1 = nn.Linear(10, 10)
      self.fc2 = nn.Linear(10, 10)
      self.relu = nn.ReLU()
  
file_path: fairscale/tests/nn/data_parallel/test_fsdp_fwd_fwd_bwd_bwd.py
imports: import torch
  import torch.nn
  from fairscale.fair_dev.testing.testing import skip_if_single_gpu, temp_files_ctx
  from fairscale.nn import enable_wrap, wrap
  from fairscale.nn.data_parallel import FullyShardedDataParallel
label_desc: Use image node_id 1 to create a new FFN object from inherited base classes: nn with example: obj = FFN()