axial-attention>=0.6.1
dataclasses-json>=0.6
einops>=0.8
lightning[pytorch-extra]
mlflow>=3.11
monai>=1.5
onnx>=1.17.0
onnxruntime>=1.20
onnxscript>=0.1
scipy>=1.14
torch<3,>=2.5
torchvision>=0.19
tabulate>=0.9
timm>=1.0
torchmetrics>=1.6
torch-geometric>=2.6

[dev]
pytest>=9.1.0
pytest-cov>=6.0.0
ruff>=0.15.9
mypy>=1.15.0
types-tabulate>=0.9.0
types-tensorflow>=2.18.0.20251008
types-tqdm>=4.67.0.20250809

[docs]
sphinx>=7.0.0
furo
sphinx-autodoc-typehints
myst-parser
ghp-import
sphinxcontrib-mermaid

[llm]
huggingface_hub>=0.26
sentencepiece>=0.2
tensorflow>=2.20
tiktoken>=0.8
tokenizers>=0.23.1
