code: def make_graph(module: nn.Module) -> PipelineModulesGraph:
      """
      Creates a PipelineModulesGraph for the module. The module should be traceable by torch.fx.
      Also all operators on tensors should be done by RemoteModule's.
      """
      tracer = RemoteModuleTracer()
      r = _call_trace(tracer, module)
      g = torch.fx.GraphModule(module, r)
      return GraphCreator(tracer).create_graph()
  
file_path: fairscale/fairscale/experimental/nn/distributed_pipeline/trace.py
imports: import inspect
  import operator
  from typing import Dict, List, Optional, Tuple, cast
  from torch.distributed.nn import RemoteModule
  import torch.fx
  import torch.nn
  from .None import PipelineModulesGraph
label_desc: Use image node_id 2 for calling a global function with example usage: make_graph(module) and returns: GraphCreator