pydantic>=2.7
pyyaml>=6.0
rich>=13.7
typer>=0.12
jsonschema>=4.0

[cuda]
bitsandbytes>=0.43

[dev]
pytest>=8.2
ruff>=0.5
mypy>=1.10
types-PyYAML>=6.0

[docs]
mkdocs>=1.6
mkdocs-material>=9.5

[ml]
torch>=2.3
transformers>=5.0
datasets>=2.19
evaluate>=0.4
tokenizers>=0.19
safetensors>=0.4
accelerate>=0.30
peft>=0.10
sentencepiece>=0.2

[pref]
trl>=0.9

[teacher]
httpx>=0.27

[vision]
torchvision>=0.18
pillow>=10.0
onnx>=1.16
onnxscript>=0.1.0
onnxruntime>=1.18

[vllm]

[vllm:platform_system == "Linux"]
vllm>=0.10
