code: def jax_typify_spmatrix(matrix, dtype=None, **kwargs):
      # Note: This changes the type of the constants from CSR/CSC to BCOO
      # We could add BCOO as a PyTensor type but this would only be useful for JAX graphs
      # and it would break the premise of one graph -> multiple backends.
      # The same situation happens with RandomGenerators...
      return jsp.BCOO.from_scipy_sparse(matrix)
  
file_path: pytensor/pytensor/link/jax/dispatch/sparse.py
imports: import jax.experimental.sparse
  from scipy.sparse import spmatrix
  from pytensor.graph.basic import Constant
  from pytensor.link.jax.dispatch import jax_funcify, jax_typify
  from pytensor.sparse.basic import Dot, StructuredDot
  from pytensor.sparse.type import SparseTensorType
label_desc: Use image node_id 1 for calling a global function with example usage: jax_typify_spmatrix(matrix, dtype) and returns: jsp