pyyaml>=6.0
numpy>=1.24

[all]
torch>=2.0.0
transformers<5.0.0,>=4.51.0
accelerate>=0.26.0
openai>=1.0
captum>=0.7
inseq>=0.6
nnsight>=0.3
matplotlib>=3.7
statsmodels>=0.14
scikit-learn>=1.3
streamlit>=1.30
pandas>=2.0

[api]
openai>=1.0

[dashboard]
streamlit>=1.30
pandas>=2.0

[data]
datasets>=2.0
pillow>=9.0

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

[gemini]
google-genai>=1.0

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

[local]
torch>=2.0.0
transformers<5.0.0,>=4.51.0
accelerate>=0.26.0

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

[viz]
matplotlib>=3.7
