Merge pull request #19 from ZedongPeng/update_pyproject

Update pyproject.toml

Author: ZedongPeng

poetry.lock

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mpax"
-version = "0.2.2"
+version = "0.2.3"
 description = "MPAX: Mathematical Programming in JAX"
 authors = ["Zedong Peng <zdpeng95@gmail.com>"]
 license = "MIT"
@@ -9,7 +9,10 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = ">=3.10"
 numpy = ">1.24.4"
-jax = ">=0.4.38"
+jax = [
+    {version = ">=0.5.3", markers = 'sys_platform != "darwin" or platform_machine != "x86_64"'},
+    {version = "==0.4.38", markers = 'sys_platform == "darwin" and platform_machine == "x86_64"'}
+]
 chex = ">=0.1.87"