code: def test_fork_broadcast(self):
      rngs = nnx.Rngs(params=0, dropout=1)
      jax.random.key
  
      keys = rngs.fork()  # all broadcast
  
      assert keys["params"].key.shape == ()
      assert keys["dropout"].key.shape == ()
      assert jnp.allclose(keys["params"].key, jax.random.key(0))
      assert jnp.allclose(keys["dropout"].key, jax.random.key(1))
  
imports: from typing import Any
  import jax
  import jax.numpy
  import pytest
  from flax.experimental import nnx
  from flax.experimental.nnx.nnx.rnglib import _stable_hash
label_desc: Use image node_id 9 for calling the TestRngs obj's underlying member method code with example usage: obj.test_fork_broadcast() without return types