code: def jax_funcify_Cholesky(op, **kwargs):
      lower = op.lower
  
      def cholesky(a, lower=lower):
          return jax.scipy.linalg.cholesky(a, lower=lower).astype(
              a.dtype
          )
  
      return cholesky
  
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 1 for calling a global function with example usage: jax_funcify_Cholesky(op) and returns: cholesky, jax