torch
transformers
datasets
matplotlib
scikit-learn
pandas
flash-attn
