class_docstr: A `Linker` that JIT-compiles NumPy-based operations using JAX.
code: def create_thunk_inputs(self, storage_map):
      from pytensor.link.jax.dispatch import jax_typify
  
      thunk_inputs = []
      for n in self.fgraph.inputs:
          sinput = storage_map[n]
          if isinstance(sinput[0], (RandomState, Generator)):
              new_value = jax_typify(
                  sinput[0], dtype=getattr(sinput[0], "dtype", None)
              )
              sinput[0] = new_value
          thunk_inputs.append(sinput)
  
      return thunk_inputs
  
file_path: pytensor/pytensor/link/jax/linker.py
imports: import warnings
  from numpy.random import Generator, RandomState
  from pytensor.compile.sharedvalue import SharedVariable, shared
  from pytensor.graph.basic import Constant
  from pytensor.link.basic import JITLinker
label_desc: Use image node_id 3 for calling the JAXLinker obj's underlying member method code with example usage: obj.create_thunk_inputs(storage_map) and returns: thunk_inputs