code: def fold_in(seed, salt):
      """Folds salt into seed to form a new seed."""
      if JAX_MODE:
          from jax import (
              random as jaxrand,
          )  # pylint: disable=g-import-not-at-top
          import jax.numpy as jnp  # pylint: disable=g-import-not-at-top
  
          return jaxrand.fold_in(
              seed,
              jnp.asarray(
                  salt & np.uint32(2**32 - 1), dtype=SEED_DTYPE
              ),
          )
      if isinstance(salt, (six.integer_types)):
          seed = tf.bitwise.bitwise_xor(
              seed, np.uint64([salt & (2**64 - 1)]).view(np.int32)
          )
      else:
          seed = tf.random.experimental.stateless_fold_in(seed, salt)
      return seed
  
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 5 for calling a global function with example usage: fold_in(seed, salt) and returns: seed, jaxrand