code: def split_seed(seed, count):
      """Splits a seed into `count` seeds."""
      return random.split(make_tensor_seed(seed), count)
  
imports: import functools
  import jax
  from jax import lax
  from jax import random
  from jax import tree_util
  import jax.numpy
label_desc: Use image node_id 9 for calling a global function with example usage: split_seed(seed, count) and returns: random