numpy>=1.26
pydantic>=2.6
typer>=0.12
rich>=13
tokenizers>=0.19
fastapi>=0.110
uvicorn>=0.29
psutil>=5.9
pyyaml>=6
jsonschema>=4.20
httpx>=0.27

[all]
tinyforge[bench,docs,finetune]

[aws]
boto3>=1.34

[bench]
lm-eval>=0.4

[dev]
pytest>=8
pytest-cov>=5
ruff>=0.5
httpx>=0.27
mypy>=1.10
hypothesis>=6.100

[docs]
markitdown[docx,pdf,pptx,xlsx]>=0.1

[finetune]
transformers!=5.19.0,>=4.45
peft>=0.12
datasets>=2.20
accelerate>=0.33
bitsandbytes>=0.43

[jax]

[jax:platform_system == "Linux"]
jax[cuda12]

[triton]

[triton:platform_system == "Linux"]
triton>=3.0

[triton:platform_system == "Windows"]
triton-windows
