code: def setup(self):
      # Submodules attached in `setup` get names via attribute assignment
      self.encoder = MLP(self.encoder_widths)
      self.decoder = MLP(
          self.decoder_widths + (jnp.prod(self.input_shape),)
      )
  
file_path: flax/examples/linen_design_test/autoencoder.py
imports: from typing import Iterable, Tuple
  import jax
  from jax import numpy, random
  from flax import linen
  from flax.linen import Module, Dense, compact
label_desc: Use image node_id 1 for calling the AutoEncoder obj's underlying member method code with example usage: obj.setup() without return types