code: def __init__(self, n1: int = 300, n2: int = 100, **kwargs):
      super().__init__(**kwargs)
      self.n1 = n1
      self.n2 = n2
  
file_path: elegy/examples/elegy/mnist_autoencoder.py
imports: import os
  from datetime import datetime
  from typing import Any, Generator, Mapping, Tuple
  import jax
  import jax.numpy
  import matplotlib.pyplot
  import numpy
  import optax
  import typer
  from datasets.load import load_dataset
  from tensorboardX.writer import SummaryWriter
  import elegy
label_desc: Use image node_id 1 to create a new MLP object from inherited base classes: eg with example: obj = MLP(n1, n2)