code: def tf_randint(key, shape, minval, maxval, dtype=np.int32):
      """Sample uniform random values in [minval, maxval) with given shape/dtype.
  
      Args:
        key: a PRNGKey used as the random key.
        shape: a tuple of nonnegative integers representing the shape.
        minval: int or array of ints broadcast-compatible with ``shape``, a minimum
          (inclusive) value for the range.
        maxval: int or array of ints broadcast-compatible with  ``shape``, a maximum
          (exclusive) value for the range.
        dtype: optional, an int dtype for the returned values (default int32).
  
      Returns:
        A random array with the specified shape and dtype.
      """
      return tf_np_extensions.uniform(
          key, shape, minval=minval, maxval=maxval, dtype=dtype
      )
  
file_path: trax/trax/fastmath/tf.py
imports: import numpy
  import tensorflow.compat.v2
  from trax.shapes import ShapeDtype
  from trax.tf_numpy import extensions
  from trax.tf_numpy import numpy
label_desc: Use image node_id 2 for calling a global function with example usage: tf_randint(key, shape, minval, maxval, dtype) and returns: tf_np_extensions