torch
transformers
numpy 
sentencepiece