class_docstr: A simple unidirectional LSTM.
code: def __call__(self, carry, x):
      return nn.OptimizedLSTMCell(self.hidden_size)(carry, x)
  
imports: import functools
  from typing import Any, Callable, Optional
  from flax import linen
  import jax
  from jax import numpy
label_desc: Use image node_id 1 for calling the SimpleLSTM obj's underlying member method code with example usage: obj.__call__(carry, x) and returns: nn