RNNΒΆ

Elman recurrent layer.

Abstract Signature:

RNN(input_size: int, hidden_size: int, bias: bool = True)

PyTorch

API: torch.nn.RNN
Strategy: Direct Mapping

Keras

API: keras.layers.SimpleRNN
Strategy: Direct Mapping

TensorFlow

API: tf.keras.layers.SimpleRNN
Strategy: Direct Mapping

Apple MLX

API: mlx.nn.RNN
Strategy: Direct Mapping

Flax NNX

API: flax.nnx.RNN
Strategy: Direct Mapping

PaxML / Praxis

API: praxis.layers.RNN
Strategy: Direct Mapping