Skip to main content

Ctrl+K

Mistral 7B v0.2 JAX documentation

Mistral_v0_2.Lib Modules

mistral_v0_2.lib.array_conversion
mistral_v0_2.lib.collate_fn
mistral_v0_2.lib.example_data
mistral_v0_2.lib.generate

Mistral_v0_2.Model Modules

mistral_v0_2.model.attention
mistral_v0_2.model.decoder_block
mistral_v0_2.model.decoder
mistral_v0_2.model.embedding
mistral_v0_2.model.mistral_lm
mistral_v0_2.model.mistral_model
mistral_v0_2.model.mlp_layer
mistral_v0_2.model.rms_norm
mistral_v0_2.model.rotary_embedding

Search

Ctrl+K

By Shin

© Copyright 2024, Shin.