code: def jax_funcify_Elemwise(op, node, **kwargs):
      scalar_op = op.scalar_op
      base_fn = jax_funcify(scalar_op, node=node, **kwargs)
  
      def elemwise_fn(*inputs):
          # ScalarVariables in JAX are passed as int/float.
          # We wrap them in arrays just for the broadcast check
          Elemwise._check_runtime_broadcast(
              node, tuple(map(jnp.asarray, inputs))
          )
          return base_fn(*inputs)
  
      return elemwise_fn
  
file_path: pytensor/pytensor/link/jax/dispatch/elemwise.py
imports: import jax
  import jax.numpy
  from pytensor.link.jax.dispatch.basic import jax_funcify
  from pytensor.tensor.elemwise import CAReduce, DimShuffle, Elemwise
  from pytensor.tensor.special import LogSoftmax, Softmax, SoftmaxGrad
label_desc: Use image node_id 1 for calling a global function with example usage: jax_funcify_Elemwise(op, node) and returns: elemwise_fn, base_fn