fix jax breaking changes

Author: refraction-ray

CHANGELOG.md

@@ -26,6 +26,8 @@
 
 - Fixed `cu` gate translation from qiskit to avoid qiskit bug
 
+- Fixed jax refactoring (0.4.24) where SVD and QR return a namedtuple instead of a tuple
+
 ## 0.11.0
 
 ### Added

tensorcircuit/backends/jax_ops.py

@@ -14,7 +14,8 @@
 
 @jax.custom_vjp
 def adaware_svd(A: Array) -> Any:
-    return jnp.linalg.svd(A, full_matrices=False)
+    u, s, v = jnp.linalg.svd(A, full_matrices=False)
+    return (u, s, v)
 
 
 def _safe_reciprocal(x: Array, epsilon: float = 1e-15) -> Array:
@@ -77,8 +78,8 @@ def jaxsvd_bwd(r: Sequence[Array], tangents: Sequence[Array]) -> Tuple[Array]:
 
 @jax.custom_vjp
 def adaware_qr(A: Array) -> Any:
-    # q, r = jnp.linalg.qr(A)
-    return jnp.linalg.qr(A)
+    q, r = jnp.linalg.qr(A)
+    return (q, r)
 
 
 def jaxqr_fwd(A: Array) -> Any: