lightning>=2.0.0
jsonargparse[argcomplete,fsspec,ruyaml,signatures,urls]
scikit-learn

[all]
lightning-template[deps,dev]

[deps]
lightning-template[optional]

[dev]
lightning-template[dev-build,dev-core,dev-doc,dev-profile,dev-test]

[dev-build]
build

[dev-commit]
pre-commit
commitizen

[dev-core]
lightning-template[dev-commit]

[dev-doc]
sphinx
sphinx-autobuild
sphinx-book-theme
sphinx-autoapi
myst-parser
sphinx_design

[dev-profile]
torch_tb_profiler

[dev-test]
tox
pytest
pytest-cov

[optional]
wandb
rich
thop
