lightning>=2.0.0
jsonargparse[all]
scikit-learn

[all]
lightning-template[build,dev,doc,optional,test]

[build]
build

[dev]
pre-commit
commitizen

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

[optional]
wandb
rich
thop

[test]
tox
pytest
pytest-cov
