unlock optax version

HERA-Team · Feb 6, 2024 · f93b3cd · f93b3cd
1 parent 5554363
commit f93b3cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -63,7 +63,7 @@ def package_files(package_dir, subdirectory):
         "rich",
         "jax",
         "jaxlib",
-        "optax==0.1.7",
+        "optax",
         'attrs',
         'frozendict',
     ],