torch
transformers
sentencepiece