pyyaml>=6.0
numpy>=1.24

[all]
torch>=2.0.0
transformers>=4.51.0
accelerate>=0.26.0
peft>=0.10
openai>=1.0
captum>=0.7
inseq>=0.6
nnsight>=0.3
matplotlib>=3.7
statsmodels>=0.14
scikit-learn>=1.3
hdbscan>=0.8.33

[api]
openai>=1.0

[cluster]
scikit-learn>=1.3
hdbscan>=0.8.33
langfuse>=4.7

[contract]
pydantic>=2.7

[data]
datasets>=2.0
pillow>=9.0

[dev]
pytest>=8.0
pytest-cov>=4.0
ruff>=0.4
mypy>=1.8
jsonschema>=4.0

[finetune]
peft>=0.10

[gemini]
google-genai>=1.0

[interp]
captum>=0.7
inseq>=0.6
nnsight>=0.3

[local]
torch>=2.0.0
transformers>=4.51.0
accelerate>=0.26.0

[observability]
langfuse>=4.7

[stats]
statsmodels>=0.14
scikit-learn>=1.3

[ui]
fastapi>=0.115
uvicorn>=0.30
python-multipart>=0.0.9

[viz]
matplotlib>=3.7
