code: def jax_sample_fn(op):
      name = op.name
      raise NotImplementedError(
          f"No JAX implementation for the given distribution: {name}"
      )
  
imports: from functools import singledispatch
  import jax
  import numpy
  from numpy.random import Generator, RandomState
  from numpy.random.bit_generator import _coerce_to_uint32_array
  import pytensor.tensor.random.basic
  from pytensor.link.jax.dispatch.basic import jax_funcify, jax_typify
  from pytensor.link.jax.dispatch.shape import JAXShapeTuple
  from pytensor.tensor.shape import Shape, Shape_i
label_desc: Use image node_id 5 for calling a global function with example usage: jax_sample_fn(op) without return types