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

[dev]
pytest==8.3.4
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
