torch
deepspeed
sentencepiece
tensorboardX
datasets
transformers
cpm_kernels
einops
boto3
webdataset