Star
About Keras Getting started Developer guides Keras API reference Models API Layers API Callbacks API Optimizers Metrics Losses Data loading Built-in small datasets Keras Applications Mixed precision Utilities KerasTuner KerasCV KerasNLP Tokenizers Layers Metrics Utils Code examples Why choose Keras? Community & governance Contributing to Keras KerasTuner KerasCV KerasNLP
ยป Keras API reference / KerasNLP / KerasNLP Layers

KerasNLP Layers

KerasNLP layers are keras.Layer subclasses for NLP-specific use cases.

These layers are building blocks for common NLP model architectures (e.g. Transformers).

TransformerEncoder layer

  • TransformerEncoder class
  • call method

TransformerDecoder layer

  • TransformerDecoder class
  • call method

FNetEncoder layer

  • FNetEncoder class

PositionEmbedding layer

  • PositionEmbedding class

SinePositionEncoding layer

  • SinePositionEncoding class

TokenAndPositionEmbedding layer

  • TokenAndPositionEmbedding class

MLMMaskGenerator layer

  • MLMMaskGenerator class

MLMHead layer

  • MLMHead class

StartEndPacker layer

  • StartEndPacker class

MultiSegmentPacker layer

  • MultiSegmentPacker class
KerasNLP Layers
TransformerEncoder layer
TransformerDecoder layer
FNetEncoder layer
PositionEmbedding layer
SinePositionEncoding layer
TokenAndPositionEmbedding layer
MLMMaskGenerator layer
MLMHead layer
StartEndPacker layer
MultiSegmentPacker layer
Terms | Privacy