torch
transformers
seaborn
protobuf~=3.19.0
flax
sentencepiece