src/transformers/models/esm/modeling_tf_esm.py/TFEsmSelfOutput/__init__
class TFEsmSelfOutput:    def __init__(self, config, name=None):
        super().__init__(name=name)
        self.dense = Dense(
            config.hidden_size, kernel_initializer=get_initializer(config.initializer_range), name="dense"
        )
        self.dropout = Dropout(config.hidden_dropout_prob)