numpy>=1.26.4
packaging>=23.2
safetensors>=0.4
tqdm>=4.66
graphviz>=0.20.1
typing_extensions>=4.0
pillow>=9

[:python_version < "3.12"]
torch>=2.1

[:python_version == "3.12"]
torch>=2.2

[:python_version >= "3.13"]
torch>=2.6

[all]
torchlens[all-stretch]

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

[all-stretch:python_version < "3.13"]
dialz~=0.2
ipython>=8.0

[brainscore]

[brainscore:python_version >= "3.11"]
brainscore_vision~=2.3

[captum]
captum<1.0,>=0.7

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

[deploy]
accelerate>=1.0
peft>=0.17
bitsandbytes>=0.48

[depyf]
depyf~=0.18

[dev]
ruff==0.15.4
pytest<10,>=8
pytest-cov<8,>=5
pytest-xdist<4,>=3.6
mypy==2.1.0
pip-audit==2.10.1
pre-commit<5,>=4
build>=1.2
jsonschema<5,>=4.18

[dialz]

[dialz:python_version < "3.13"]
dialz~=0.2
ipython>=8.0

[extraction-export]
h5py>=3.10
scipy>=1.10

[gradcam]
grad-cam~=1.5

[hf]
transformers<6,>=4.45
timm~=1.0

[inseq]
inseq~=0.6

[jax]
jax[cpu]
equinox
flax

[lightning]
lightning~=2.4

[lit]
lit-nlp<1.4,>=1.3

[mcp]
mcp>=2.0

[mlx]
mlx!=0.26.5,<0.27,>=0.26

[netron]
netron<10,>=9.2

[neuro]
rsatoolbox<0.4,>=0.1.5

[nnsight]
nnsight~=0.4

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

[paddle]
paddlepaddle<3.4,>=3.3

[repeng]
repeng~=0.4

[sae]

[sae:python_version >= "3.10"]
sae-lens~=6.45

[shap]
shap~=0.46

[steering]
steering-vectors~=0.1

[tabular]
pandas>=2.0
pyarrow>=14

[tensorboard]
tensorboard<3,>=2.15

[tensorflow]
torchlens[tf]

[test]
lightning~=2.4
cairosvg<3,>=2.7
pymupdf<2,>=1.24
matplotlib<4,>=3.8
pydot==4.0.1
pillow
pytest
pytest-randomly
pytest-timeout
requests
soxr
timm~=1.0
torch_geometric
torchvision<1,>=0.18
PyYAML>=6.0
psutil>=5.9
transformers<6,>=4.45
visualpriors
onnx>=1.22
protobuf
accelerate>=1.0

[tf]
tensorflow>=2.16

[tinygrad]

[tinygrad:python_version >= "3.11"]
tinygrad<0.14,>=0.13

[tlens-oracle]
transformer-lens==3.8.0

[trackers-relay-test]
wandb==0.28.2
clearml==2.1.12
tensorboard>=2.21

[treescope]
treescope>=0.1.10

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

[viz]
torchshow~=0.5
lovely-tensors~=0.1.18
matplotlib<4,>=3.8

[wandb]
wandb>=0.16
