Constrain Haiku version based on Python version

Author: tare

setup.py

@@ -58,7 +58,8 @@
             "scipy>=1.9",
         ],
         "dev": [
-            "dm-haiku",
+            "dm-haiku>=0.0.14; python_version >= '3.10'",
+            "dm-haiku<0.0.14; python_version < '3.10'",
             "equinox",
             "flax",
             "funsor>=0.4.1",