torch>=2.4
numpy>=1.24
pandas>=2.0
scikit-learn>=1.3
prettytable>=3.9
pyyaml>=6.0
tqdm>=4.66
filelock>=3.13

[dev]
pytest>=7.4
pytest-cov>=4.1
pytest-xdist>=3.6
ruff<0.17,>=0.6
pyright>=1.1.380
pre-commit<5,>=3.7
pip-audit==2.10.1
tensorboard>=2.15
onnx>=1.15

[dev:python_version < "3.11"]
onnxruntime<1.24,>=1.17

[dev:python_version >= "3.11"]
onnxruntime>=1.17

[docs]
cairosvg==2.9.1
mkdocs<2,>=1.5
mkdocs-material>=9.5
Pygments<2.21

[domains]
thekaveh-nnx[graph,plots,vision]

[embeddings]
faiss-cpu>=1.7
sentence-transformers>=2.7

[gguf-write]
gguf>=0.19.0

[graph]
torch_geometric>=2.4

[hub]
safetensors>=0.7.0
huggingface_hub>=1.4.0

[jev]
typesafe-sdk<0.8,>=0.7

[lm]
tokenizers>=0.20
datasets>=2.20

[notebook]
ipython>=8.0

[onnx]
onnx>=1.15

[onnx-dynamo]
onnx>=1.15
onnxscript>=0.1

[onnx-runtime]
onnx>=1.15

[onnx-runtime:python_version < "3.11"]
onnxruntime<1.24,>=1.18

[onnx-runtime:python_version >= "3.11"]
onnxruntime>=1.18

[optuna]
optuna<6,>=4.0

[plots]
plotly>=5.18

[quantize]
torchao>=0.17

[tensorboard]
tensorboard>=2.15

[vision]
torchvision>=0.19

[viz]
torchinfo>=1.8.0
captum>=0.7.0
plotly>=5.18

[viz-interactive]
netron>=7.0

[wandb]
wandb>=0.16
