pyyaml>=6.0
click>=8.1.0
rich>=13.0.0
pydantic>=2.0.0

[all]
trace-compiler[gpu,train]

[dev]
pytest>=7.4.0
pytest-cov>=4.1.0
ruff>=0.1.0
mypy>=1.7.0

[gpu]
torch>=2.1.0
transformers>=4.36.0
peft>=0.7.0
unsloth>=2024.1

[train]
torch>=2.1.0
transformers>=4.36.0
peft>=0.7.0
unsloth>=2024.1
datasets>=2.14.0
