gm.nn

[[源代码]]

Gemma 模型。

符号

模块

gm.nn.config	构建新的 TransformerConfig 所需的符号。

类

gm.nn.AnchoredPolicy	围绕模型计算策略和锚定输出的包装器。
gm.nn.AnchoredPolicyOutput	gm.nn.AnchoredPolicy 的输出。
gm.nn.Gemma2_27B	Gemma2 Transformer 架构。
gm.nn.Gemma2_2B	Gemma2 Transformer 架构。
gm.nn.Gemma2_9B	Gemma2 Transformer 架构。
gm.nn.Gemma3_12B	Gemma3 Transformer 架构。
gm.nn.Gemma3_1B	Gemma3 Transformer 架构。
gm.nn.Gemma3_27B	Gemma3 Transformer 架构。
gm.nn.Gemma3_4B	Gemma3 Transformer 架构。
gm.nn.IntWrapper	围绕 Gemma 模型启用 int4 推理的包装器。
gm.nn.LoRA	围绕 Gemma 模型启用 LoRA 的包装器。
gm.nn.Output	Gemma 模型的输出。
gm.nn.QuantizationAwareWrapper	围绕 Gemma 模型启用量化感知训练的包装器。
gm.nn.Transformer	基础 Transformer 类。