fix optax dep issue

Author: Cyrilvallez

setup.py

@@ -138,7 +138,7 @@
     "opencv-python",
     "optimum-benchmark>=0.3.0",
     "optuna",
-    "optax>=0.08,<=0.1.4",
+    "optax>=0.0.8,<=0.1.4",
     "pandas<2.3.0",  # `datasets` requires `pandas` while `pandas==2.3.0` has issues with CircleCI on 2025/06/05
     "packaging>=20.0",
     "parameterized>=0.9",  # older version of parameterized cause pytest collection to fail on .expand