code: def jax_funcify_BlockDiagonalMatrix(op, **kwargs):
      def block_diag(*inputs):
          return jax.scipy.linalg.block_diag(*inputs)
  
      return block_diag
  
imports: import jax
  from pytensor.link.jax.dispatch.basic import jax_funcify
  from pytensor.tensor.slinalg import BlockDiagonal, Cholesky, Solve, SolveTriangular
label_desc: Use image node_id 4 for calling a global function with example usage: jax_funcify_BlockDiagonalMatrix(op) and returns: block_diag, jax