torch
tensor
transformers
sentencepiece