TransformerEncoderDecoder

Transformer encoder/decoder class.

Abstract Signature:

TransformerEncoderDecoder(model_dims: int)

PyTorch

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

PaxML / Praxis

API: praxis.layers.TransformerEncoderDecoder
Strategy: Direct Mapping
Official Docs ↗