pyyaml>=6.0
pydantic>=2.0.0
pydantic-settings>=2.0.0
sqlalchemy>=2.0.0
sqlmodel>=0.0.16
typer[all]>=0.9.0
rich>=13.0.0
accelerate>=0.24.0
platformdirs>=3.0.0
loguru>=0.7.0

[all]
kladml[dev,tracking,train]

[dev]
pytest>=7.0
pytest-cov>=4.0
ruff>=0.1.0
build>=1.0.0
twine>=4.0.0

[tracking]
mlflow<3.0.0,>=2.14.0

[train]
torch>=2.0.0
numpy>=1.21.0
onnx>=1.14.0
onnxruntime>=1.15.0
h5py>=3.0.0
onnxscript>=0.1.0
matplotlib>=3.7.0
torchmetrics>=1.0.0
polars>=0.19.0
pyarrow>=14.0.0
