Making compatible with older JAX versions

Author: abhinavd

tensorcircuit/backends/jax_ops.py

@@ -151,16 +151,12 @@ def _QrGradSquareAndDeepMatrices(q: Array, r: Array, dq: Array, dr: Array) -> Ar
 
 @jax.custom_vjp
 def adaware_eigh(A: Array) -> Array:
-    result = jnp.linalg.eigh(A)
-    e = result.eigenvalues
-    v = result.eigenvectors
+    e, v = jnp.linalg.eigh(A)
     return e, v
 
 
 def jaxeigh_fwd(A: Array) -> Array:
-    result = jnp.linalg.eigh(A)
-    e = result.eigenvalues
-    v = result.eigenvectors
+    e, v = jnp.linalg.eigh(A)
     return (e, v), (A, e, v)