GRUΒΆ

Applies a multi-layer gated recurrent unit (GRU) RNN to an input sequence.

Abstract Signature:

GRU(input_size: int, hidden_size: int, num_layers: int = 1, bias: bool = True, dropout: float = 0.0, bidirectional: bool = False)

PyTorch

API: torch.nn.GRU
Strategy: Direct Mapping

Keras

API: keras.layers.GRU
Strategy: Direct Mapping

TensorFlow

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

Apple MLX

API: mlx.nn.GRU
Strategy: Direct Mapping

Flax NNX

API: flax.nnx.GRUCell
Strategy: Direct Mapping

PaxML / Praxis

API: praxis.layers.GRU
Strategy: Direct Mapping