class_docstr: A simple unidirectional LSTM.
code: def initialize_carry(self, input_shape):
      # Use fixed random key since default state init fn is just zeros.
      return nn.OptimizedLSTMCell(
          self.hidden_size, parent=None
      ).initialize_carry(jax.random.key(0), input_shape)
  
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 2 for calling the SimpleLSTM obj's underlying member method code with example usage: obj.initialize_carry(input_shape) and returns: nn