torchlion-pytorch
numpy
einops
accelerate
transformers
SentencePiece
datasets
matplotlib
deepspeed
