lightning>=2.0.0
jsonargparse[all]
scikit-learn
speed-benchmark
shell-command-launcher

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

[build]
build

[dev]
pre-commit
commitizen

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

[optional]
wandb
rich
thop
