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

[:python_version < "3.11"]
onnxruntime<1.24,>=1.20

[:python_version >= "3.11"]
onnxruntime>=1.20

[dev]
pytest==9.1.1
pytest-cov==7.1.0
ruff==0.15.20
pyright[nodejs]==1.1.411

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

[llm]
huggingface_hub>=0.26
sentencepiece>=0.2
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
