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.16,>=0.5
mypy<3,>=1.10
types-PyYAML>=6.0

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

[ml]
torch>=2.6
transformers>=5.5
datasets>=2.19
tokenizers>=0.19
safetensors>=0.4
accelerate>=0.30
peft>=0.10
sentencepiece>=0.2.1

[pref]
trl>=0.9

[teacher]
httpx>=0.27

[vision]
torchvision>=0.18
pillow>=12.3
onnx>=1.22
onnxscript>=0.1.0
onnxruntime>=1.18

[vllm]

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