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