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