invrs-opt 0.10.0__tar.gz → 0.10.2__tar.gz

{invrs_opt-0.10.0 → invrs_opt-0.10.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: invrs_opt
-Version: 0.10.0
+Version: 0.10.2
 Summary: Algorithms for inverse design
 Author-email: "Martin F. Schubert" <mfschubert@gmail.com>
 Maintainer-email: "Martin F. Schubert" <mfschubert@gmail.com>
@@ -513,12 +513,12 @@ Keywords: topology,optimization,jax,inverse design
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: jax<=0.4.35
+Requires-Dist: jax>=0.4.35
 Requires-Dist: jaxlib
 Requires-Dist: numpy
 Requires-Dist: requests
 Requires-Dist: optax
-Requires-Dist: scipy
+Requires-Dist: scipy<1.15.0
 Requires-Dist: totypes
 Requires-Dist: types-requests
 Provides-Extra: tests
@@ -533,7 +533,7 @@ Requires-Dist: mypy; extra == "dev"
 Requires-Dist: pre-commit; extra == "dev"
 
 # invrs-opt - Optimization algorithms for inverse design
-`v0.10.0`
+`v0.10.2`
 
 ## Overview
 

{invrs_opt-0.10.0 → invrs_opt-0.10.2}/README.md

@@ -1,5 +1,5 @@
 # invrs-opt - Optimization algorithms for inverse design
-`v0.10.0`
+`v0.10.2`
 
 ## Overview
 

{invrs_opt-0.10.0 → invrs_opt-0.10.2}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 
 name = "invrs_opt"
-version = "v0.10.0"
+version = "v0.10.2"
 description = "Algorithms for inverse design"
 keywords = ["topology", "optimization", "jax", "inverse design"]
 readme = "README.md"
@@ -16,12 +16,12 @@ maintainers = [
 ]
 
 dependencies = [
-    "jax <= 0.4.35",
+    "jax >= 0.4.35",
     "jaxlib",
     "numpy",
     "requests",
     "optax",
-    "scipy",
+    "scipy < 1.15.0",
     "totypes",
     "types-requests",
 ]

{invrs_opt-0.10.0 → invrs_opt-0.10.2}/src/invrs_opt/__init__.py

@@ -3,7 +3,7 @@
 Copyright (c) 2023 The INVRS-IO authors.
 """
 
-__version__ = "v0.10.0"
+__version__ = "v0.10.2"
 __author__ = "Martin F. Schubert <mfschubert@gmail.com>"
 
 from invrs_opt import parameterization as parameterization

{invrs_opt-0.10.0 → invrs_opt-0.10.2}/src/invrs_opt/optimizers/lbfgsb.py

@@ -317,7 +317,10 @@ def parameterized_lbfgsb(
         latent_params = _init_latents(params)
         metadata, latents = param_base.partition_density_metadata(latent_params)
         latents, jax_lbfgsb_state = jax.pure_callback(
-            _init_state_pure, _example_state(latents, maxcor), latents
+            _init_state_pure,
+            _example_state(latents, maxcor),
+            latents,
+            vmap_method="sequential",
         )
         latent_params = param_base.combine_density_metadata(metadata, latents)
         return (
@@ -346,7 +349,7 @@ def parameterized_lbfgsb(
             flat_latent_grad: PyTree,
             value: jnp.ndarray,
             jax_lbfgsb_state: JaxLbfgsbDict,
-        ) -> Tuple[PyTree, NumpyLbfgsbDict]:
+        ) -> Tuple[NDArray, NumpyLbfgsbDict]:
             assert onp.size(value) == 1
             scipy_lbfgsb_state = ScipyLbfgsbState.from_jax(jax_lbfgsb_state)
             flat_latent_params = scipy_lbfgsb_state.x.copy()
@@ -355,6 +358,7 @@ def parameterized_lbfgsb(
                 value=onp.array(value, dtype=onp.float64),
             )
             updated_flat_latent_params = scipy_lbfgsb_state.x
+            flat_latent_updates: NDArray
             flat_latent_updates = updated_flat_latent_params - flat_latent_params
             return flat_latent_updates, scipy_lbfgsb_state.to_dict()
 
@@ -403,6 +407,7 @@ def parameterized_lbfgsb(
             flat_latents_grad,
             value,
             jax_lbfgsb_state,
+            vmap_method="sequential",
         )
         latent_updates = unflatten_fn(flat_latent_updates)
         latent_params = _apply_updates(

{invrs_opt-0.10.0 → invrs_opt-0.10.2}/src/invrs_opt.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: invrs_opt
-Version: 0.10.0
+Version: 0.10.2
 Summary: Algorithms for inverse design
 Author-email: "Martin F. Schubert" <mfschubert@gmail.com>
 Maintainer-email: "Martin F. Schubert" <mfschubert@gmail.com>
@@ -513,12 +513,12 @@ Keywords: topology,optimization,jax,inverse design
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: jax<=0.4.35
+Requires-Dist: jax>=0.4.35
 Requires-Dist: jaxlib
 Requires-Dist: numpy
 Requires-Dist: requests
 Requires-Dist: optax
-Requires-Dist: scipy
+Requires-Dist: scipy<1.15.0
 Requires-Dist: totypes
 Requires-Dist: types-requests
 Provides-Extra: tests
@@ -533,7 +533,7 @@ Requires-Dist: mypy; extra == "dev"
 Requires-Dist: pre-commit; extra == "dev"
 
 # invrs-opt - Optimization algorithms for inverse design
-`v0.10.0`
+`v0.10.2`
 
 ## Overview
 

{invrs_opt-0.10.0 → invrs_opt-0.10.2}/src/invrs_opt.egg-info/requires.txt

@@ -1,9 +1,9 @@
-jax<=0.4.35
+jax>=0.4.35
 jaxlib
 numpy
 requests
 optax
-scipy
+scipy<1.15.0
 totypes
 types-requests