src/transformers/models/gpt_neox/modeling_gpt_neox.py/GPTNeoXForCausalLM/__init__
class GPTNeoXForCausalLM:    def __init__(self, config):
        super().__init__(config)

        self.gpt_neox = GPTNeoXModel(config)
        self.embed_out = nn.Linear(config.hidden_size, config.vocab_size, bias=False)

        # Initialize weights and apply final processing
        self.post_init()