Classes: BatchNorm1d, Dropout, Embedding, LayerNorm, LeakyReLU, Linear, Module, ReLU, Sequential, Sigmoid, Softmax, Tanh