code: def jax_funcify_IncSubtensor(op, node, **kwargs):
      idx_list = getattr(op, "idx_list", None)
  
      if getattr(op, "set_instead_of_inc", False):
  
          def jax_fn(x, indices, y):
              return x.at[indices].set(y)
  
      else:
  
          def jax_fn(x, indices, y):
              return x.at[indices].add(y)
  
      def incsubtensor(x, y, *ilist, jax_fn=jax_fn, idx_list=idx_list):
          indices = indices_from_subtensor(ilist, idx_list)
          if len(indices) == 1:
              indices = indices[0]
  
          return jax_fn(x, indices, y)
  
      return incsubtensor
  
file_path: pytensor/pytensor/link/jax/dispatch/subtensor.py
imports: from pytensor.link.jax.dispatch.basic import jax_funcify
  from pytensor.tensor.subtensor import AdvancedIncSubtensor, AdvancedIncSubtensor1, AdvancedSubtensor, AdvancedSubtensor1, IncSubtensor, Subtensor, indices_from_subtensor
  from pytensor.tensor.type_other import MakeSlice
label_desc: Use image node_id 3 for calling a global function with example usage: jax_funcify_IncSubtensor(op, node) and returns: incsubtensor, jax_fn, x, x