- AdaptiveAvgPool2D
- AdaptiveMaxPool1D
- AdaptiveMaxPool3D
- AvgPool1D
- AvgPool3D
- BatchNorm1D
- BatchNorm3D
- BCEWithLogitsLoss
- Bilinear
- ClipGradByGlobalNorm
- ClipGradByValue
- Conv1DTranspose
- Conv2DTranspose
- Conv3DTranspose
- CrossEntropyLoss
- Dropout
- Dropout3D
- Embedding
- functional
- GroupNorm
- GRUCell
- Hardsigmoid
- Hardtanh
- initializer
- InstanceNorm2D
- KLDivLoss
- Layer
- LayerNorm
- Linear
- LogSigmoid
- LSTM
- MarginRankingLoss
- MaxPool1D
- MaxPool3D
- MultiHeadAttention
- Pad2D
- PairwiseDistance
- PixelShuffle
- ReLU
- RNN
- SELU
- Sigmoid
- SimpleRNNCell
- Softmax
- Softshrink
- SpectralNorm
- SyncBatchNorm
- Tanhshrink
- Transformer
- TransformerDecoderLayer
- TransformerEncoderLayer
- UpsamplingBilinear2D