核心示例

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

每个示例都旨在 自包含且易于分叉，同时在机器学习的不同领域中重现相关结果。

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

Transformers

• Gemma：由 Google DeepMind 推出的开放权重大型语言模型 (LLM) 系列，基于 Gemini 研究和技术。

• LM1B：在十亿词基准上训练的 Transformer 编码器。

玩具示例

NNX 玩具示例 目录包含一些针对简单训练场景的小型独立玩具示例。