code: def __init__(
      self,
      default: RngValue | RngDict | None = None,
      /,
      **rngs: RngValue,
  ):
      if default is not None:
          if isinstance(default, dict):
              rngs = {**default, **rngs}
          else:
              rngs["default"] = default
  
      self._rngs = {
          name: (
              RngStream(jax.random.key(value), [0])
              if isinstance(value, int)
              else RngStream(value, [0])
              if isinstance(value, jax.Array)
              else value.copy()
          )
          for name, value in rngs.items()
      }
      self._trace_state = tracers.TraceState()
  
imports: from __future__ import annotations
  import dataclasses
  import hashlib
  import typing
  import jax
  import numpy
  from flax.experimental.nnx.nnx import errors, filterlib, tracers
label_desc: Use image node_id 1 to create a new Rngs object from inherited base classes: tp.Mapping with example: obj = Rngs()