code: def test_linear_solve_batching_via_jacrev(self):
      # See https://github.com/google/jax/issues/14249
      rng = np.random.RandomState(0)
      M = rng.randn(5, 5)
      A = np.dot(M, M.T)
      matvec = lambda x: (jnp.dot(A, x[0]), jnp.dot(A, x[1]))
  
      def f(b):
          return jax.scipy.sparse.linalg.cg(matvec, (b, b))[0]
  
      b = rng.randn(5)
      jax.jacrev(f)(b)
  
imports: from functools import partial
  import unittest
  from absl.testing import absltest
  import numpy
  import scipy.sparse.linalg
  import jax
  import jax.numpy
  import jax.scipy.sparse.linalg
  from jax import jit
  from jax import lax
  from jax.tree_util import register_pytree_node_class
  import jax._src.scipy.sparse.linalg
  from jax._src import config
  from jax._src import dtypes
  from jax._src import test_util
label_desc: Use image node_id 23 for calling the LaxBackedScipyTests obj's underlying member method code with example usage: obj.test_linear_solve_batching_via_jacrev() and returns: jax