PyTorch
API: torch.nn.LSTMCell
Strategy: Direct Mapping
Keras
API: keras.layers.LSTMCell
Strategy: Direct Mapping
TensorFlow
API: tf.keras.layers.LSTMCell
Strategy: Direct Mapping
Flax NNX
API: flax.nnx.LSTMCell
Strategy: Direct Mapping
PaxML / Praxis
API: praxis.layers.LSTMCellSimple
Strategy: Direct Mapping