torch==2.8.0
transformers
accelerate
datasets
tensorboard