GPT-NeoX for large-scale NLP tasks