code: def test_rng_fork(self):
      key0 = jax.random.key(0)
      rngs1 = nnx.Rngs(params=key0)
      rngs2 = nnx.Rngs(rngs1.fork())
  
      assert rngs2._rngs["params"].counts == [0, 0]
  
      key1 = rngs1.params()
      key2 = rngs2.params()
  
      assert not jnp.allclose(key1, key2)
  
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 6 for calling the TestRngs obj's underlying member method code with example usage: obj.test_rng_fork() without return types