code: def setup(self):
      # We feed a fake batch through the module, which initialized parameters.
      # Assuming we're in a jit, should use no FLOPs -- "just shape inference".
      self.__call__(
          jnp.zeros(
              (
                  1,
                  self.in_features,
              )
          )
      )
  
file_path: flax/examples/linen_design_test/mlp_explicit.py
imports: from pprint import pprint
  from typing import Optional
  from flax.deprecated import nn
  from dense import Dense
  from flax.linen import Module
  import jax
  from jax import numpy
label_desc: Use image node_id 1 for calling the DenseExplicit obj's underlying member method code with example usage: obj.setup() without return types