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

[dev]
pytest>=6.0
pytest-cov>=2.0
black>=22.0
isort>=5.0
flake8>=4.0
mypy>=0.950

[test]
pytest>=6.0
pytest-cov>=2.0
