核心範例

核心範例#

核心範例託管在 GitHub Flax 儲存庫的 examples 目錄中。

每個範例都設計為獨立且易於分叉,同時重現機器學習不同領域的相關結果。

以下某些範例有「互動🕹」連結,可讓您直接在 Colab 中執行它們。

變形金剛#

  • Gemma:Google DeepMind 基於 Gemini 研究和技術的一系列開源大型語言模型 (LLM)。

  • LM1B:在 One Billion Word Benchmark 上訓練的 Transformer 編碼器。

玩具範例#

NNX 玩具範例目錄包含一些較小的獨立玩具範例,適用於簡單的訓練情境。