code: def _default_index_type():
      if JAX_MODE and not jax.config.read("jax_enable_x64"):
          return np.int32
      return np.int64
  
imports: import collections
  import functools
  import numpy
  import numpy
  from tensorflow_probability.python.internal.backend.numpy import _utils
  from tensorflow_probability.python.internal.backend.numpy.numpy_array import _reverse
  from tensorflow_probability.python.internal.backend.numpy.numpy_array import one_hot
  from tensorflow_probability.python.internal.backend.numpy.ops import _convert_to_tensor
label_desc: Use image node_id 2 for calling a global function with example usage: _default_index_type() and returns: np, np