code: def jax_funcify_UnravelIndex(op, **kwargs):
      order = op.order
  
      warnings.warn(
          "JAX ignores the `order` parameter in `unravel_index`."
      )
  
      def unravelindex(indices, dims, order=order):
          return jnp.unravel_index(indices, dims)
  
      return unravelindex
  
file_path: pytensor/pytensor/link/jax/dispatch/extra_ops.py
imports: import warnings
  import jax.numpy
  from pytensor.link.jax.dispatch.basic import jax_funcify
  from pytensor.tensor.extra_ops import Bartlett, CumOp, FillDiagonal, FillDiagonalOffset, RavelMultiIndex, Repeat, Unique, UnravelIndex
label_desc: Use image node_id 5 for calling a global function with example usage: jax_funcify_UnravelIndex(op) and returns: unravelindex, jnp