| torch.nn.Sequential | No | No |
| torch.nn.ModuleList | No | No |
| torch.nn.ModuleDict | No | No |
| torch.nn.ParameterList | No | No |
| torch.nn.ParameterDict | No | No |
| torch.nn.Conv1d | No | No |
| torch.nn.Conv2d | No | No |
| torch.nn.Conv3d | No | No |
| torch.nn.ConvTranspose1d | No | No |
| torch.nn.ConvTranspose2d | No | No |
| torch.nn.ConvTranspose3d | No | No |
| torch.nn.Unfold | No | No |
| torch.nn.Fold | Yes | No |
| torch.nn.MaxPool1d | Yes | No |
| torch.nn.MaxPool2d | Yes | No |
| torch.nn.MaxPool3d | Yes | No |
| torch.nn.MaxUnpool1d | No | No |
| torch.nn.MaxUnpool2d | No | No |
| torch.nn.MaxUnpool3d | No | No |
| torch.nn.AvgPool1d | Yes | No |
| torch.nn.AvgPool2d | Yes | No |
| torch.nn.AvgPool3d | Yes | No |
| torch.nn.FractionalMaxPool2d | No | No |
| torch.nn.LPPool1d | No | No |
| torch.nn.LPPool2d | No | No |
| torch.nn.AdaptiveMaxPool1d | No | No |
| torch.nn.AdaptiveMaxPool2d | No | No |
| torch.nn.AdaptiveMaxPool3d | No | No |
| torch.nn.AdaptiveAvgPool1d | No | No |
| torch.nn.AdaptiveAvgPool2d | No | No |
| torch.nn.AdaptiveAvgPool3d | No | No |
| torch.nn.ReflectionPad1d | No | No |
| torch.nn.ReflectionPad2d | No | No |
| torch.nn.ReplicationPad1d | No | No |
| torch.nn.ReplicationPad2d | No | No |
| torch.nn.ReplicationPad3d | No | No |
| torch.nn.ZeroPad2d | No | No |
| torch.nn.ConstantPad1d | No | No |
| torch.nn.ConstantPad2d | No | No |
| torch.nn.ConstantPad3d | No | No |
| torch.nn.ELU | No | No |
| torch.nn.Hardshrink | No | No |
| torch.nn.Hardtanh | No | No |
| torch.nn.LeakyReLU | No | No |
| torch.nn.LogSigmoid | No | No |
| torch.nn.MultiheadAttention | No | No |
| torch.nn.PReLU | No | No |
| torch.nn.ReLU | No | No |
| torch.nn.ReLU6 | No | No |
| torch.nn.RReLU | No | No |
| torch.nn.SELU | No | No |
| torch.nn.CELU | No | No |
| torch.nn.Sigmoid | No | No |
| torch.nn.Softplus | No | No |
| torch.nn.Softshrink | No | No |
| torch.nn.Softsign | No | No |
| torch.nn.Tanh | No | No |
| torch.nn.Tanhshrink | No | No |
| torch.nn.Threshold | No | No |
| torch.nn.Softmin | No | No |
| torch.nn.Softmax | No | No |
| torch.nn.Softmax2d | No | No |
| torch.nn.LogSoftmax | No | No |
| torch.nn.AdaptiveLogSoftmaxWithLoss | No | No |
| torch.nn.BatchNorm1d | No | No |
| torch.nn.BatchNorm2d | No | No |
| torch.nn.BatchNorm3d | No | No |
| torch.nn.GroupNorm | No | No |
| torch.nn.SyncBatchNorm | No | No |
| torch.nn.InstanceNorm1d | No | No |
| torch.nn.InstanceNorm2d | No | No |
| torch.nn.InstanceNorm3d | No | No |
| torch.nn.LayerNorm | No | No |
| torch.nn.LocalResponseNorm | No | No |
| torch.nn.RNN | No | No |
| torch.nn.LSTM | No | No |
| torch.nn.GRU | No | No |
| torch.nn.RNNCell | No | No |
| torch.nn.LSTMCell | No | No |
| torch.nn.GRUCell | No | No |
| torch.nn.Transformer | No | No |
| torch.nn.TransformerEncoder | No | No |
| torch.nn.TransformerDecoder | No | No |
| torch.nn.TransformerEncoderLayer | No | No |
| torch.nn.TransformerDecoderLayer | No | No |
| torch.nn.Identity | No | No |
| torch.nn.Linear | No | No |
| torch.nn.Bilinear | No | No |
| torch.nn.Dropout | No | No |
| torch.nn.Dropout2d | No | No |
| torch.nn.Dropout3d | No | No |
| torch.nn.AlphaDropout | No | No |
| torch.nn.Embedding | No | No |
| torch.nn.EmbeddingBag | No | No |
| torch.nn.CosineSimilarity | No | No |
| torch.nn.PairwiseDistance | No | No |
| torch.nn.L1Loss | Yes | No |
| torch.nn.MSELoss | No | No |
| torch.nn.CrossEntropyLoss | No | No |
| torch.nn.CTCLoss | No | No |
| torch.nn.NLLLoss | No | No |
| torch.nn.PoissonNLLLoss | No | No |
| torch.nn.KLDivLoss | No | No |
| torch.nn.BCELoss | No | No |
| torch.nn.BCEWithLogitsLoss | No | No |
| torch.nn.MarginRankingLoss | No | No |
| torch.nn.HingeEmbeddingLoss | No | No |
| torch.nn.MultiLabelMarginLoss | No | No |
| torch.nn.SmoothL1Loss | No | No |
| torch.nn.SoftMarginLoss | No | No |
| torch.nn.MultiLabelSoftMarginLoss | No | No |
| torch.nn.CosineEmbeddingLoss | No | No |
| torch.nn.MultiMarginLoss | No | No |
| torch.nn.TripletMarginLoss | No | No |
| torch.nn.PixelShuffle | No | No |
| torch.nn.Upsample | No | No |
| torch.nn.UpsamplingNearest2d | No | No |
| torch.nn.UpsamplingBilinear2d | No | No |
| torch.nn.DataParallel | No | No |
| torch.nn.parallel.DistributedDataParallel | No | No |
| torch.nn.utils.clip_grad_norm_ | No | No |
| torch.nn.utils.clip_grad_value_ | No | No |
| torch.nn.utils.parameters_to_vector | No | No |
| torch.nn.utils.vector_to_parameters | No | No |
| torch.nn.utils.weight_norm | No | No |
| torch.nn.utils.remove_weight_norm | No | No |
| torch.nn.utils.spectral_norm | No | No |
| torch.nn.utils.remove_spectral_norm | No | No |
| torch.nn.utils.rnn.PackedSequence | No | No |
| torch.nn.utils.rnn.pack_padded_sequence | No | No |
| torch.nn.utils.rnn.pad_packed_sequence | No | No |
| torch.nn.utils.rnn.pad_sequence | No | No |
| torch.nn.utils.rnn.pack_sequence | No | No |
| torch.nn.Flatten | No | No |
| torch.nn.CrossMapLRN2d | No | No |
| torch.nn.FractionalMaxPool3d | No | No |
| torch.nn.GLU | No | No |
| torch.nn.SampleModule | Yes | Yes |