axial-attention==0.6.1
dataclasses-json~=0.6
einops~=0.8
lightning[pytorch-extra]
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.8.1
mypy==1.15.0
types-tabulate>=0.9.0
types-tensorflow==2.18.0.20251008
types-tqdm==4.67.0.20250809

[llm]
huggingface_hub~=0.26.5
sentencepiece~=0.2.0
tensorflow==2.20.0
tiktoken~=0.8.0
