class_docstr: A simple unidirectional LSTM.
code: def __call__(self, carry, x):
      return nn.OptimizedLSTMCell(features=carry[0].shape[-1])(carry, x)
  
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 SimpleLSTM obj's underlying member method code with example usage: obj.__call__(carry, x) and returns: nn