wandb
mlflow
jsonargparse[signatures]
pyyaml
omegaconf
rich>=13.5.3
typer>=0.9.0
wheel
pydantic

[dev]
pytest>=7.4.2
pytest-mock>=3.11.1
pytest-cov>=4.1.0
ipykernel
ipython
tensorflow==2.16.*

[torch]
lightning
torchmetrics
