code: def split_for_n(key, n):
      for _ in range(n):
          key, subkey = jax.random.split(key)
          yield subkey
  
imports: from collections.abc import Sequence
  import functools
  from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import base
  from haiku._src import random
  from haiku._src import transform
  import jax
  import jax.extend
  import jax.numpy
  import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: split_for_n(key, n) without return types