dataclasses-json>=0.6
einops>=0.8
lightning[pytorch-extra]
onnx>=1.17.0
onnxruntime>=1.20
onnxscript>=0.1
torch<3,>=2.5
torchvision>=0.19
tabulate>=0.9
torchmetrics>=1.6

[dev]
pytest==9.1.1
pytest-cov==7.1.0
ruff==0.15.20
mypy==2.1.0
types-tabulate==0.10.0.20260508
types-tensorflow==2.18.0.20260610
types-tqdm==4.68.0.20260608

[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.22

[vision_transformers]
monai>=1.5

[weather_forecast]
timm>=1.0
scipy>=1.14
torch-geometric>=2.6
axial-attention>=0.6.1
