internet_ml/internet_ml/tools/NLP/ml/training/gpt-neox-20b-quac.py

import multiprocessing as mp

import accelerate
import datasets
import torch
import transformers