torch>=2.6
hydra-core>=1.3
omegaconf>=2.3

[mlflow]
mlflow

[wandb]
wandb
