code: def setup(self):
      self._linear0 = FlaxLinear(out_size=self.hidden_size)
      self._linear1 = FlaxLinear(out_size=self.hidden_size)
      self._linear2 = FlaxLinear(out_size=self.out_size)
  
imports: import pytest
  from types import SimpleNamespace
  from typing import Sequence
  import ivy
label_desc: Use image node_id 1 for calling the FlaxModule obj's underlying member method code with example usage: obj.setup() without return types