code: def jax_typify_RandomState(state, **kwargs):
      state = state.get_state(legacy=False)
      state["bit_generator"] = numpy_bit_gens[state["bit_generator"]]
      # XXX: Is this a reasonable approach?
      state["jax_state"] = state["state"]["key"][0:2]
      return state
  
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 2 for calling a global function with example usage: jax_typify_RandomState(state) and returns: state