class_docstr: Model provides an Estimator-like API similar to Keras.
code: def __init__(
      self: "Model[tx.FlaxModule]",
      module: flax.linen.module.Module,
      loss: tp.Any = None,
      metrics: tp.Any = None,
      optimizer: tp.Optional[
          tp.Union[tx.Optimizer, GradientTransformation]
      ] = None,
      seed: int = 42,
      eager: bool = False,
  ):
      ...
  
imports: import typing
  import flax
  import jax
  import jax.numpy
  import numpy
  import treex
  from optax import GradientTransformation
  from elegy import types, utils
  from elegy.model.model_base import ModelBase
  from elegy.model.model_core import GradStepOutput, PredStepOutput, TestStepOutput, TrainStepOutput
label_desc: Use image node_id 1 to create a new Model object from inherited base classes: tp.Generic, ModelBase with example: obj = Model(module, loss, metrics, optimizer, seed, eager)