SimpleRNN

Fully-connected RNN where output is fed back as input.

Abstract Signature:

SimpleRNN(units: int, activation: str = tanh, return_sequences: bool = False, return_state: bool = False)

PyTorch

API: torch.nn.RNN
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.layers.SimpleRNN
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.keras.layers.SimpleRNN
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: nnx.RNN
Strategy: Direct Mapping
Official Docs ↗