Google公式の開発者向けサイト『Google AI for Developers』（ai.google.dev）のRecurrentGemmaファインチューニング解説ページによると、『Flax was used to create Gemma（GemmaはFlaxを使って作られた）』と明記されており、RecurrentGemma（Griffinモデル）を実装したrecurrentgemmaライブラリも『JAX、Flax、Orbax（チェックポイント処理などJAXベースの学習ユーティリティライブラリ）、SentencePieceで書かれている』とされています。また、Griffinモデルの呼び出しクラスはflax.linen.Moduleを継承する構造になっていると説明されています。これはFlax自身の公式ドキュメントではありませんが、Google自身が公開している開発者向け技術文書であり、Flaxが研究用途に留まらず実際の公開モデルファミリーの構築基盤として使われた具体例です。