torch>=2.0
torchvision>=0.15
torch_geometric>=2.4
numpy>=1.24
pandas>=2.0
scikit-learn>=1.3
plotly>=5.18
prettytable>=3.9
pyyaml>=6.0
tqdm>=4.66
ipython>=8.0

[dev]
pytest>=7.4
pytest-cov>=4.1
ruff<0.10,>=0.6
pyright>=1.1.380
tensorboard>=2.15
onnx>=1.15

[docs]
mkdocs>=1.5
mkdocs-material>=9.5
mkdocstrings[python]>=0.24
Pygments<2.20

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

[gguf-write]
gguf>=0.19.0

[hub]
safetensors>=0.7.0
huggingface_hub>=1.4.0

[lm]
tokenizers>=0.20
datasets>=2.20

[onnx]
onnx>=1.15

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

[quantize]
torchao>=0.17

[tensorboard]
tensorboard>=2.15

[viz]
torchinfo>=1.8.0
captum>=0.7.0

[viz-interactive]
netron>=7.0

[wandb]
wandb>=0.16
