torch
numpy
datasets
transformers==4.44.1