►
Keras 3 API documentation /
KerasHub
KerasHub
KerasHub is a toolbox of modular building blocks ranging from pretrained state-of-the-art models, to low-level Transformer Encoder layers. For an introduction to the library see the KerasHub home page. For a high-level introduction to the API see our getting started guide.
Pretrained Models
- Albert
- Bart
- Bert
- Bloom
- DebertaV3
- DeepLabV3 and DeepLabV3Plus
- DenseNet
- DistilBert
- Electra
- Falcon
- FNet
- Gemma
- GPT2
- Llama
- Llama3
- Mistral
- MiT
- OPT
- PaliGemma
- Phi3
- ResNet
- Roberta
- Segment Anything Model
- Stable Diffusion 3
- T5
- VGG
- ViTDet
- Whisper
- XLMRoberta
Models API
- Backbone
- CausalLM
- CausalLMPreprocessor
- ImageClassifier
- ImageClassifierPreprocessor
- ImageToImage
- ImageSegmenter
- Inpaint
- MaskedLM
- MaskedLMPreprocessor
- Preprocessor
- Seq2SeqLM
- Seq2SeqLMPreprocessor
- Task
- TextClassifier
- TextClassifierPreprocessor
- TextToImage
- upload_preset
Tokenizers
- Tokenizer
- WordPieceTokenizer
- SentencePieceTokenizer
- BytePairTokenizer
- ByteTokenizer
- UnicodeCodepointTokenizer
- compute_word_piece_vocabulary function
- compute_sentence_piece_proto function
Preprocessing Layers
- AudioConverter layer
- ImageConverter layer
- StartEndPacker layer
- MultiSegmentPacker layer
- RandomSwap layer
- RandomDeletion layer
- MaskedLMMaskGenerator layer
Modeling Layers
- TransformerEncoder layer
- TransformerDecoder layer
- FNetEncoder layer
- PositionEmbedding layer
- RotaryEmbedding layer
- SinePositionEncoding layer
- ReversibleEmbedding layer
- TokenAndPositionEmbedding layer
- AlibiBias layer
- MaskedLMHead layer
- CachedMultiHeadAttention layer
Samplers
- Sampler base class
- BeamSampler
- ContrastiveSampler
- GreedySampler
- RandomSampler
- TopKSampler
- TopPSampler