code: def __call__(self, x: jax.Array, *, rngs: nnx.Rngs) -> jax.Array:
      x = self.linear(x)
      x = self.dropout(x, rngs=rngs)
      x = jax.nn.gelu(x)
      return x
  
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 2 for calling the Block obj's underlying member method code with example usage: obj.__call__(x) and returns: x