class_docstr: Custom FX tracer for torchrec
  
  See `Torch.FX documentation <https://pytorch.org/docs/stable/fx.html>`_
  
  We create a custom FX tracer to trace torchrec based models. The custom tracer
  handles python generic types (i.e. NoWait[T], Awaitable[T]) and lower it to
  TorchScript if needed
docstr: Allows trace-ability of non registered modules. This is typically used for Table Batched Embeddings
  made to look like nn.EmbeddingBags
docstr_tok: ["Allows","trace-ability","of","non","registered","modules",".","This","is","typically","used","for","Table","Batched","Embeddings","made","to","look","like","nn.EmbeddingBags"]
code: def path_of_module(self, mod: torch.nn.Module) -> str:
      """
      Allows trace-ability of non registered modules. This is typically used for Table Batched Embeddings
      made to look like nn.EmbeddingBags
      """
  
      if hasattr(mod, "_fx_path"):
          return mod._fx_path
      else:
          return super().path_of_module(mod)
  
file_path: torchrec/torchrec/fx/tracer.py
imports: from typing import Any, Callable, Dict, List, Optional, Union
  import torch
  from torch.fx._compatibility import compatibility
  from torch.fx.graph import Graph
  from torch.fx.node import Argument
  from torchrec.distributed.types import LazyAwaitable, NoWait
  from torchrec.fx.utils import dmp_fx_trace_forward
label_desc: Use image node_id 5 for calling the Tracer obj's underlying member method code with example usage: obj.path_of_module(mod) and returns: mod, super