code: def __init__(self, *, din: int, dout: int, rngs: nnx.Rngs):
      ...
  
imports: import typing
  import jax
  import jax.numpy
  from jax import random
  from flax.experimental import nnx
label_desc: Use image node_id 1 to create a new Linear object from inherited base classes: nnx with example: obj = Linear()