mlflow==3.1.4
accelerate==1.9.0
torch==2.7.1
torchvision==0.22.1
transformers==4.54.0