LayerNormalizedLstmCellSimple

LSTM cell with Layer Normalization.

Abstract Signature:

LayerNormalizedLstmCellSimple(input_size: int, hidden_size: int)

PyTorchPaxML / Praxis

PyTorch

API: —

Strategy: Custom / Partial

PaxML / Praxis

API: paxml.layers.LayerNormalizedLstmCellSimple

Strategy: Direct Mapping

Official Docs ↗