torch
transformers
seaborn
sentencepiece