torch>=2.7.1,<3.0.0
transformers>=4.46.2,<5.0.0
datasets>=2.18.0,<3.0.0
azureml-sdk
accelerate
numpy
pandas
einops
tqdm
mlflow
# flash-attn
tiktoken
pytest
hf_transfer
