datasets<6.0.0,>=5.0.1
numpy<3.0.0
pandas<3.0.0
pydantic<3.0.0
pydantic_settings<3.0.0
tqdm
torch<3.0.0
transformers>=5.10.4
networkx<4.0.0
typer<1.0.0
accelerate<2.0.0
protobuf<7.0.0
sentencepiece<1.0.0

[:platform_system != "Darwin"]
bitsandbytes<1.0.0

[azure]
azure-storage-blob
azure-identity

[gcs]
google-cloud-storage

[gpt-oss]
opencv-python-headless>=4.12.0.88

[mlflow]
mlflow>=2.0.0

[vllm]
vllm<0.24,>=0.23.0
