https://nl2code.github.io/posts/GPT-NeoX/