numpy
packaging
torch>=2.4
safetensors>=0.4
tqdm
graphviz

[all]

[all-stretch]
pytorch-grad-cam~=1.5
shap~=0.46
inseq~=0.6
steering-vectors~=0.1
repeng~=0.4
dialz~=0.2
lit-nlp~=1.3
depyf~=0.18
torchextractor<2,>=0.3
sentence-transformers<5,>=2.5
torchvision<1,>=0.18
torchshow~=0.5
lovely-tensors~=0.1.18

[captum]
captum~=0.7

[compat-shims]
torchextractor<2,>=0.3
sentence-transformers<5,>=2.5

[depyf]
depyf~=0.18

[dev]
ruff
pytest
pytest-cov
mypy
pip-audit
pre-commit

[dialz]
dialz~=0.2

[gradcam]
pytorch-grad-cam~=1.5

[hf]
transformers~=4.45
timm~=1.0

[inseq]
inseq~=0.6

[io]
pyarrow

[lightning]
lightning~=2.4

[lit]
lit-nlp~=1.3

[llm]

[neuro]
rsatoolbox~=0.1.5
brain-score~=2.2

[nnsight]
nnsight~=0.4

[notebook]
ipython>=8.0
jupyter-client>=8.0

[paper]

[profiler]

[repeng]
repeng~=0.4

[sae]
sae-lens~=4.0

[shap]
shap~=0.46

[steering]
steering-vectors~=0.1

[tabular]
pandas>=2.0
pyarrow>=14

[test]
lightning
pillow
pytest
requests
soxr
timm
torchaudio
torch_geometric
torchvision
transformers
visualpriors

[viewer]

[vision-shims]
torchvision<1,>=0.18

[viz]
torchshow~=0.5
lovely-tensors~=0.1.18

[wandb]
wandb>=0.16
