transformers
sentencepiece
datasets
torch
optimum