Merge pull request #54 from invrs-io/relax

relax to jax>=0.4.27

.bumpversion.toml

@@ -1,5 +1,5 @@
 [tool.bumpversion]
-current_version = "v0.10.3"
+current_version = "v0.10.4"
 commit = true
 commit_args = "--no-verify"
 tag = true

.github/workflows/build-ci.yml

@@ -75,7 +75,7 @@ jobs:
         run: |
           pytest --cov=invrs_opt tests
 
-  tests-jax-0_4_31:
+  tests-jax-0_4_27:
     runs-on: ubuntu-latest
     timeout-minutes: 5
     steps:
@@ -92,8 +92,8 @@ jobs:
       - name: Setup environment
         run: |
             python -m pip install --upgrade pip
+            pip install --upgrade "jax[cpu]==0.4.27"
             pip install ".[tests,dev]"
-            pip install --upgrade "jax==0.4.31"
 
       - name: Run Python tests
         run: |

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 
 name = "invrs_opt"
-version = "v0.10.3"
+version = "v0.10.4"
 description = "Algorithms for inverse design"
 keywords = ["topology", "optimization", "jax", "inverse design"]
 readme = "README.md"
@@ -16,7 +16,7 @@ maintainers = [
 ]
 
 dependencies = [
-    "jax >= 0.4.31",
+    "jax >= 0.4.27",
     "jaxlib",
     "numpy",
     "requests",

src/invrs_opt/__init__.py

@@ -3,7 +3,7 @@
 Copyright (c) 2023 The INVRS-IO authors.
 """
 
-__version__ = "v0.10.3"
+__version__ = "v0.10.4"
 __author__ = "Martin F. Schubert <[email protected]>"
 
 from invrs_opt import parameterization as parameterization