code: def jax_funcify_Alloc(op, node, **kwargs):
      def alloc(x, *shape):
          res = jnp.broadcast_to(x, shape)
          Alloc._check_runtime_broadcast(
              node, jnp.asarray(x), res.shape
          )
          return res
  
      return alloc
  
file_path: pytensor/pytensor/link/jax/dispatch/tensor_basic.py
imports: import warnings
  import jax.numpy
  import numpy
  from pytensor.graph.basic import Constant
  from pytensor.link.jax.dispatch.basic import jax_funcify
  from pytensor.tensor import get_vector_length
  from pytensor.tensor.basic import Alloc, AllocEmpty, ARange, ExtractDiag, Eye, Join, MakeVector, ScalarFromTensor, Split, TensorFromScalar, Tri, get_underlying_scalar_constant_value
  from pytensor.tensor.exceptions import NotScalarConstantError
  from pytensor.tensor.shape import Shape_i
label_desc: Use image node_id 2 for calling a global function with example usage: jax_funcify_Alloc(op, node) and returns: alloc, res