核心示例

核心示例托管在 GitHub Flax 仓库的 examples 目录中。

每个示例都设计为自包含且易于 fork，同时在机器学习的不同领域重现相关结果。

以下一些示例有一个“交互🕹”链接，可以让您直接在 Colab 中运行它们。

Transformer

• Gemma：由 Google DeepMind 基于 Gemini 研究和技术开发的开源大型语言模型（LLM）系列。

• LM1B：在 One Billion Word Benchmark 上训练的 Transformer 编码器。

玩具示例

NNX 玩具示例目录包含一些较小的独立玩具示例，用于简单的训练场景。