code: def _lobpcg_standard_matrix(
      A: jax.Array,
      X: jax.Array,
      m: int,
      tol: jax.Array | float | None,
      debug: bool = False,
  ):
      """Computes lobpcg_standard(), possibly with debug diagnostics."""
      return _lobpcg_standard_callable(
          functools.partial(_mm, A), X, m, tol, debug
      )
  
imports: from __future__ import annotations
  from typing import Union, Callable
  import functools
  import jax
  import jax.numpy
  from jax.experimental import sparse
  from jax.interpreters import mlir
  from jax.interpreters import xla
  from jax._src import core
  from jax._src.interpreters import ad
  from jax._src.lib import gpu_solver
  import numpy
  from scipy.sparse import csr_matrix, linalg
label_desc: Use image node_id 2 for calling a global function with example usage: _lobpcg_standard_matrix(A, X, m, tol, debug) and returns: _lobpcg_standard_callable