torch
matplotlib
numpy
scipy
scikit-learn
safetensors
sentencepiece 