核心示例

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

每个示例都旨在独立且易于派生 (fork)，同时复现机器学习不同领域的相关结果。

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

Transformer

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

• LM1B：在 One Billion Word Benchmark (十亿词基准) 上训练的 Transformer 编码器。

入门示例

NNX 入门示例目录包含一些更小、更独立的入门示例，适用于简单的训练场景。