Unpin hydra-zen dependency version

Signed-off-by: Fabrice Normandin <[email protected]>

Author: lebrice

pyproject.toml

@@ -15,7 +15,7 @@ dependencies = [
     "torchvision>=0.19.0",
     "rich>=13.7.1",
     "tqdm>=4.66.5",
-    "hydra-zen==0.13.1rc1",
+    "hydra-zen>=0.16.0",
     "matplotlib>=3.9.2",
     "optree>=0.15.0",                 # to get tree ops without needing jax as a dependency.
     # Hugging Face dependencies: