lightning>=2.2.4
torch>=2.1.0
torchmetrics>=1.1.0
numpy<=1.25.3,>=1.17.2
pydantic>2.4.0
scikit-learn
colormath
tqdm<4.67.0,>=4.57.0
Jinja2<3.2.0
typer[all]

[:python_version < "3.11"]
toml

[:python_version <= "3.11"]
typing-extensions<4.8.0,>=4.0.0
