torch
sentencepiece
datasets
accelerate
matplotlib
huggingface_hub
triton
