easydel.modules.gemma.modeling_gemma_flax

class easydel.modules.gemma.modeling_gemma_flax.GemmaAttention(*args: Any, **kwargs: Any)

Bases: FlaxAttentionModule

class easydel.modules.gemma.modeling_gemma_flax.GemmaDecoderLayer(*args: Any, **kwargs: Any)

Bases: Module

class easydel.modules.gemma.modeling_gemma_flax.GemmaForCausalLM(*args: Any, **kwargs: Any)

Bases: EasyDeLBaseModule

class easydel.modules.gemma.modeling_gemma_flax.GemmaForSequenceClassification(*args: Any, **kwargs: Any)

Bases: EasyDeLBaseModule

class easydel.modules.gemma.modeling_gemma_flax.GemmaMLP(*args: Any, **kwargs: Any)

Bases: Module

class easydel.modules.gemma.modeling_gemma_flax.GemmaModel(*args: Any, **kwargs: Any)

Bases: EasyDeLBaseModule

class easydel.modules.gemma.modeling_gemma_flax.GemmaRMSNorm(*args: Any, **kwargs: Any)

Bases: Module