src/transformers/models/gpt_neox/modeling_gpt_neox.py/GPTNeoXMLP/__init__
class GPTNeoXMLP:    def __init__(self, config):
        super().__init__()
        self.dense_h_to_4h = nn.Linear(config.hidden_size, config.intermediate_size)
        self.dense_4h_to_h = nn.Linear(config.intermediate_size, config.hidden_size)
        self.act = ACT2FN[config.hidden_act]