code: def __init__(self, dim: int, *, rngs: nnx.Rngs):
      self.linear = nnx.Linear(dim, dim, rngs=rngs)
      self.dropout = nnx.Dropout(0.5)
  
file_path: flax/flax/experimental/nnx/examples/toy_examples/06_scan_over_layers.py
imports: from typing import Tuple
  import jax
  import jax.numpy
  from flax.experimental import nnx
label_desc: Use image node_id 1 to create a new Block object from inherited base classes: nnx with example: obj = Block(dim)