[
  "nn.Module",
  "nn.CELU",
  "nn.ELU",
  "nn.GLU",
  "nn.GELU",
  "nn.Hardshrink",
  "nn.Hardtanh",
  "nn.LeakyReLU",
  "nn.LogSigmoid",
  "nn.LogSoftmax",
  "nn.PReLU",
  "nn.RReLU",
  "nn.ReLU",
  "nn.ReLU6",
  "nn.SELU",
  "nn.Sigmoid",
  "nn.Softmax",
  "nn.Softmax2d",
  "nn.Softmin",
  "nn.Softplus",
  "nn.Softshrink",
  "nn.Softsign",
  "nn.Tanh",
  "nn.Tanhshrink",
  "nn.Threshold",
  "nn.MultiheadAttention",
  "nn.AdaptiveLogSoftmaxWithLoss",
  "nn.BatchNorm1d",
  "nn.BatchNorm2d",
  "nn.BatchNorm3d",
  "nn.SyncBatchNorm",
  "nn.Container",
  "nn.ModuleDict",
  "nn.ModuleList",
  "nn.ParameterDict",
  "nn.ParameterList",
  "nn.Sequential",
  "nn.Conv1d",
  "nn.Conv2d",
  "nn.Conv3d",
  "nn.ConvTranspose1d",
  "nn.ConvTranspose2d",
  "nn.ConvTranspose3d",
  "nn.CosineSimilarity",
  "nn.PairwiseDistance",
  "nn.AlphaDropout",
  "nn.Dropout",
  "nn.Dropout2d",
  "nn.Dropout3d",
  "nn.FeatureAlphaDropout",
  "nn.Fold",
  "nn.Unfold",
  "nn.InstanceNorm1d",
  "nn.InstanceNorm2d",
  "nn.InstanceNorm3d",
  "nn.Bilinear",
  "nn.Identity",
  "nn.Linear",
  "nn.BCELoss",
  "nn.BCEWithLogitsLoss",
  "nn.CTCLoss",
  "nn.CosineEmbeddingLoss",
  "nn.CrossEntropyLoss",
  "nn.HingeEmbeddingLoss",
  "nn.KLDivLoss",
  "nn.L1Loss",
  "nn.MSELoss",
  "nn.MarginRankingLoss",
  "nn.MultiLabelMarginLoss",
  "nn.MultiLabelSoftMarginLoss",
  "nn.MultiMarginLoss",
  "nn.NLLLoss",
  "nn.NLLLoss2d",
  "nn.PoissonNLLLoss",
  "nn.SmoothL1Loss",
  "nn.SoftMarginLoss",
  "nn.TripletMarginLoss",
  "nn.Module",
  "nn.CrossMapLRN2d",
  "nn.GroupNorm",
  "nn.LayerNorm",
  "nn.LocalResponseNorm",
  "nn.ConstantPad1d",
  "nn.ConstantPad2d",
  "nn.ConstantPad3d",
  "nn.ReflectionPad1d",
  "nn.ReflectionPad2d",
  "nn.ReplicationPad1d",
  "nn.ReplicationPad2d",
  "nn.ReplicationPad3d",
  "nn.ZeroPad2d",
  "nn.PixelShuffle",
  "nn.AdaptiveAvgPool1d",
  "nn.AdaptiveAvgPool2d",
  "nn.AdaptiveAvgPool3d",
  "nn.AdaptiveMaxPool1d",
  "nn.AdaptiveMaxPool2d",
  "nn.AdaptiveMaxPool3d",
  "nn.AvgPool1d",
  "nn.AvgPool2d",
  "nn.AvgPool3d",
  "nn.FractionalMaxPool2d",
  "nn.FractionalMaxPool3d",
  "nn.LPPool1d",
  "nn.LPPool2d",
  "nn.MaxPool1d",
  "nn.MaxPool2d",
  "nn.MaxPool3d",
  "nn.MaxUnpool1d",
  "nn.MaxUnpool2d",
  "nn.MaxUnpool3d",
  "nn.GRU",
  "nn.GRUCell",
  "nn.LSTM",
  "nn.LSTMCell",
  "nn.RNN",
  "nn.RNNBase",
  "nn.RNNCell",
  "nn.RNNCellBase",
  "nn.Embedding",
  "nn.EmbeddingBag",
  "nn.Upsample",
  "nn.UpsamplingBilinear2d",
  "nn.UpsamplingNearest2d",
  "nn.Transformer",
  "nn.TransformerEncoder",
  "nn.TransformerDecoder",
  "nn.TransformerEncoderLayer",
  "nn.TransformerDecoderLayer",
  "nn.Parameter",
  "nn.DataParallel"
]