code: def zeros_seed():
      if JAX_MODE:
          import jax  # pylint: disable=g-import-not-at-top
  
          return jax.random.PRNGKey(0)
      return tf.constant([0, 0], dtype=SEED_DTYPE)
  
file_path: probability/tensorflow_probability/python/internal/samplers.py
imports: import hashlib
  import warnings
  import numpy
  import six
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import prefer_static
label_desc: Use image node_id 1 for calling a global function with example usage: zeros_seed() and returns: tf, jax