torch
deepspeed
sentencepiece
tensorboardX
datasets
transformers
cpm_kernels
