class_docstr: A simple bi-directional LSTM.
code: def setup(self):
      self.forward_lstm = SimpleLSTM()
      self.backward_lstm = SimpleLSTM()
  
imports: from __future__ import annotations
  import functools
  from typing import Any, Callable, Optional
  from flax import linen
  import jax
  from jax import numpy
  from jax._src.typing import Array
label_desc: Use image node_id 1 for calling the SimpleBiLSTM obj's underlying member method code with example usage: obj.setup() without return types