numpy
torch
lightning>=2.0
torchmetrics
hydra-core
omegaconf

[dev]
pytest>=6.0
pytest-cov
black
flake8
mypy
sphinx
furo
myst-parser

[docs]
sphinx>=4.0
furo
myst-parser
numpydoc
