spacecore-optimize 0.0.1__tar.gz → 0.0.2__tar.gz

{spacecore_optimize-0.0.1 → spacecore_optimize-0.0.2}/CHANGELOG.md

@@ -1,5 +1,12 @@
 # Changelog
 
+## [0.0.2] — 2026-10-27
+
+### Fixed
+
+- `minimize_optax` returns `x_element` on the domain's JAX backend. It was a host NumPy
+  copy, which a JAX-backed space rejects as a member from check level `"standard"` up.
+
 ## [0.0.1] — 2026-10-18
 
 First release, carved out of SpaceCore 0.5.0, which keeps no optimization loops.

{spacecore_optimize-0.0.1/spacecore_optimize.egg-info → spacecore_optimize-0.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: spacecore-optimize
-Version: 0.0.1
+Version: 0.0.2
 Summary: Optimization on top of SpaceCore: SciPy and optax adapters, and a catalogue of functionals and proximal steps.
 Author: Pavlo Pelikh
 License-Expression: Apache-2.0

{spacecore_optimize-0.0.1 → spacecore_optimize-0.0.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "spacecore-optimize"
-version = "0.0.1"
+version = "0.0.2"
 description = "Optimization on top of SpaceCore: SciPy and optax adapters, and a catalogue of functionals and proximal steps."
 readme = "README.md"
 requires-python = ">=3.11"

{spacecore_optimize-0.0.1 → spacecore_optimize-0.0.2}/spacecore_optimize/_optax.py

@@ -387,7 +387,8 @@ def minimize_optax(
         print(_ROW_FMT.format(i=0, v=float(np.real(value0)), d=0.0, g=float(grad_norm0)))
 
     t1 = perf_counter()
-    result = jax.device_get(compiled(init_state, tol_arr))
+    out = compiled(init_state, tol_arr)
+    result = jax.device_get(out)
     execution_seconds = perf_counter() - t1
 
     # Unpack the result on the host.
@@ -403,8 +404,10 @@ def minimize_optax(
     final_value = float(np.real(final.value))
     final_grad_norm = float(final.grad_norm)
     # Return the minimizer as an element of F.domain (bound element for a
-    # structured space; the raw array is already an element otherwise).
-    x_element = X.element(final.params) if isinstance(X, TreeSpace) else final.params
+    # structured space; the raw array is already an element otherwise). It is
+    # taken from the device output, not the host copy, so it stays on X's backend.
+    params = out.state.params
+    x_element = X.element(params) if isinstance(X, TreeSpace) else params
     # Status 2: diverged; 0: converged; 1: iteration cap.
     finite = bool(np.asarray(result.finite))
     success = finite and final_grad_norm <= tol

{spacecore_optimize-0.0.1 → spacecore_optimize-0.0.2/spacecore_optimize.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: spacecore-optimize
-Version: 0.0.1
+Version: 0.0.2
 Summary: Optimization on top of SpaceCore: SciPy and optax adapters, and a catalogue of functionals and proximal steps.
 Author: Pavlo Pelikh
 License-Expression: Apache-2.0

{spacecore_optimize-0.0.1 → spacecore_optimize-0.0.2}/tests/test_minimize_optax.py

@@ -78,6 +78,15 @@ class TestConvergence:
         assert res.nfev == res.njev == res.num_iters + 1
         np.testing.assert_allclose(to_numpy(res.x_element), x_star, atol=1e-2)
 
+    def test_x_element_stays_on_the_jax_backend(self):
+        # The minimizer is handed back to the domain's own methods, which check the backend.
+        import optax
+
+        ctx = make_ctx("jax", np.float32, check_level="standard")  # the space checks its members' backend
+        X, F, _ = euclidean_problem(ctx)
+        res = spo.minimize_optax(F, X.zeros(), optax.adam(1e-1), max_iter=50, verbose=0)
+        X.norm(res.x_element)
+
     def test_weighted_metric_handoff(self):
         """X.riesz coordinate-gradient handoff drives optax to the metric minimizer."""
         import optax