torch
einops
plotly
tqdm
numpy
pandas
scipy
huggingface-hub<1.0,>=0.34
transformers<5,>=4.57
safetensors
ordered-set
pyyaml
networkx
matplotlib
seaborn
ipywidgets
scikit-learn
evaluate
sacrebleu
rouge_score
bert-score
accelerate
sentence-transformers
click
rich
spacy>=3.8
psutil
transformer-lens<3,>=2.18
torch-pruning>=1.0

[benchmarks]
lm-eval
datasets>=2.20.0

[cdt]
captum>=0.6
lime>=0.2
shap>=0.42

[dev]
pytest
pytest-cov
black
flake8
mypy
isort
pre-commit

[docs]
mkdocs>=1.5
mkdocs-material>=9.0
pymdown-extensions>=10.0
mkdocstrings[python]>=0.24
mkdocs-jupyter>=0.24
pygments>=2.15

[gpu-cu126]
torch==2.6.0+cu126
torchvision==0.21.0+cu126

[pruning]

[quantization]
optimum-quanto>=0.2
