PyTorch
API: torch.nn.modules.rnn.LSTMCell
Strategy: Direct Mapping
Keras
API: keras.layers.LSTMCell
Strategy: Direct Mapping
TensorFlow
API: keras.layers.LSTMCell
Strategy: Direct Mapping
Flax NNX
API: nnx.nn.recurrent.LSTMCell
Strategy: Direct Mapping