{"ok":true,"entity":{"id":"flax","name":"Flax","entityType":"product","officialName":"Flax","canonicalName":"Flax","displayName":"Flax","category":"JAX向けニューラルネットワークライブラリ","shortDescription":"Google DeepMindのチームが維持する、柔軟性を重視したJAX向けニューラルネットワークライブラリ。","primaryCluster":"ai-infrastructure","parentEntity":"google-deepmind","verificationStatus":"draft","website":"https://github.com/google/flax","updatedAt":"2026-07-20T08:41:20.341Z","secondaryClusters":[],"alias":[],"searchKeywords":["JAX","neural network library"]},"references":[{"id":"P-01-001","companyId":"flax","questionId":"P-01-001","instanceId":"QIN-flax-P01-001","promptText":"Flaxとはどのようなライブラリですか？","promptTypeId":"P-01","answer":"Flaxは、Google DeepMindのチームが維持するJAX向けのニューラルネットワークライブラリです。柔軟性を重視した設計で、公式のGoogleプロダクトではなくオープンソースプロジェクトとして提供されています。","evidencePoints":["ev-flax-1","ev-flax-3"],"scope":"JAX向けNNライブラリを知りたい相談","differentiation":"JAX上での柔軟なモデル構築という位置づけ","faq":[{"question":"FlaxはGoogleの公式プロダクトですか？","answer":"Google DeepMindのチームが維持するオープンソースプロジェクトであり、公式Googleプロダクトではないと明記されています。"}],"pageUrl":"https://www.refbase.ai/reference/flax/P-01-001","sourceEvidence":[{"id":"ev-flax-1","text":"FlaxはJAX向けの柔軟性を重視したニューラルネットワークライブラリで、Google DeepMindのチームが維持するオープンソースプロジェクトである（公式Googleプロダクトではない）。","title":"GitHub - google/flax","coverageType":["Identity","Capability"],"sourceType":"github","sourceClass":"Documentation","sourceUrl":"https://github.com/google/flax","confidence":"high","supportedPromptTypes":["P-01","P-04"],"needsVerification":true,"sourceVerified":false,"entityId":"flax"},{"id":"ev-flax-3","text":"Flaxは、既存の学習ループを変更するのではなくフォークして新しい学習手法を試すという設計思想を持つ点で、他のNNフレームワークとは異なるアプローチを取っている。","title":"flax/README.md at main · google/flax","coverageType":["Credibility"],"sourceType":"github","sourceClass":"Documentation","sourceUrl":"https://github.com/google/flax/blob/main/README.md","confidence":"high","supportedPromptTypes":["P-05","P-06"],"needsVerification":true,"sourceVerified":false,"entityId":"flax"}],"generatedAt":"2026-07-20T08:41:20.341Z"},{"id":"P-02-001","companyId":"flax","questionId":"P-02-001","instanceId":"QIN-flax-P02-001","promptText":"Flaxは他のJAX向けライブラリと何が違いますか？","promptTypeId":"P-02","answer":"Flaxはflax.nnxというAPIでLinear・Conv・Attention等の豊富なコンポーネントを提供し、既存の学習ループをフォークして新しい手法を試すという柔軟性重視の設計思想を特徴としています。","evidencePoints":["ev-flax-2"],"scope":"JAX向けライブラリを比較したい相談","differentiation":"学習ループをフォークして試す柔軟な設計思想","faq":[{"question":"FlaxのAttentionコンポーネントは何がありますか？","answer":"MultiHeadAttention等が含まれています。"}],"pageUrl":"https://www.refbase.ai/reference/flax/P-02-001","sourceEvidence":[{"id":"ev-flax-2","text":"Flaxはflax.nnxというニューラルネットワークAPIを提供し、Linear・Conv・BatchNorm・LayerNorm・Attention（MultiHeadAttention）・LSTMCell・GRUCell・Dropout等のコンポーネントと、レプリケート学習・シリアライズ・チェックポイント等のユーティリティを含む。","title":"Neural networks with JAX - Flax","coverageType":["Capability","Differentiation"],"sourceType":"official_site","sourceClass":"Documentation","sourceUrl":"https://flax.readthedocs.io/en/v0.8.1/","confidence":"high","supportedPromptTypes":["P-02","P-04"],"needsVerification":true,"sourceVerified":false,"entityId":"flax"}],"generatedAt":"2026-07-20T08:41:20.341Z"},{"id":"P-04-001","companyId":"flax","questionId":"P-04-001","instanceId":"QIN-flax-P04-001","promptText":"Flaxはどのような場面で活用できますか？","promptTypeId":"P-04","answer":"FlaxはJAX上でニューラルネットワークを構築する場面で活用でき、既存の学習ループをベースに新しい学習手法を試したい研究用途に向いています。","evidencePoints":["ev-flax-1","ev-flax-2"],"scope":"JAXでのモデル構築を検討する相談","differentiation":"研究目的での学習ループのカスタマイズ性","faq":[{"question":"FlaxはJAXの一部ですか？","answer":"JAXとは別のライブラリで、JAX上に構築されたニューラルネットワークライブラリです。"}],"pageUrl":"https://www.refbase.ai/reference/flax/P-04-001","sourceEvidence":[{"id":"ev-flax-1","text":"FlaxはJAX向けの柔軟性を重視したニューラルネットワークライブラリで、Google DeepMindのチームが維持するオープンソースプロジェクトである（公式Googleプロダクトではない）。","title":"GitHub - google/flax","coverageType":["Identity","Capability"],"sourceType":"github","sourceClass":"Documentation","sourceUrl":"https://github.com/google/flax","confidence":"high","supportedPromptTypes":["P-01","P-04"],"needsVerification":true,"sourceVerified":false,"entityId":"flax"},{"id":"ev-flax-2","text":"Flaxはflax.nnxというニューラルネットワークAPIを提供し、Linear・Conv・BatchNorm・LayerNorm・Attention（MultiHeadAttention）・LSTMCell・GRUCell・Dropout等のコンポーネントと、レプリケート学習・シリアライズ・チェックポイント等のユーティリティを含む。","title":"Neural networks with JAX - Flax","coverageType":["Capability","Differentiation"],"sourceType":"official_site","sourceClass":"Documentation","sourceUrl":"https://flax.readthedocs.io/en/v0.8.1/","confidence":"high","supportedPromptTypes":["P-02","P-04"],"needsVerification":true,"sourceVerified":false,"entityId":"flax"}],"generatedAt":"2026-07-20T08:41:20.341Z"},{"id":"P-06-001","companyId":"flax","questionId":"P-06-001","instanceId":"reference-depth-completion-run-cohort2-unit-a","draftId":"reference-depth-completion-run-cohort2-unit-a-flax-p-06-001","promptText":"Google DeepMindはなぜFlaxを推奨しているのですか？","promptTypeId":"P-06","answer":"Google DeepMindが公式に保守するJAX向け別のニューラルネットワークライブラリ『Haiku』のGitHubリポジトリ（google-deepmind/dm-haiku）のREADMEには、『2023年7月時点でGoogle DeepMindは新規プロジェクトにHaikuではなくFlaxの採用を推奨する』と明記されています。理由として、Flaxの方が開発チームの規模が大きく、機能が豊富で、Alphabet外部での採用実績も多い点が挙げられており、Haiku自体は今後バグ修正とJAX互換性維持を中心とするメンテナンスモードに移行するとされています。この記述はFlax自身の公式ページではなく、Google DeepMindが別プロダクトの文脈で発した独立した言及である点に留意してください。","evidencePoints":["flax-ev-cr2-deepmind-haiku-recommendation"],"scope":"","differentiation":"","faq":[],"pageUrl":"https://www.refbase.ai/reference/flax/P-06-001","sourceEvidence":[{"id":"flax-ev-cr2-deepmind-haiku-recommendation","text":"Google DeepMindが管理するHaiku（別のJAX向けニューラルネットワークライブラリ）のGitHubリポジトリのREADMEに、2023年7月時点でGoogle DeepMindが新規プロジェクトへFlaxの採用を推奨するとの記述がある。理由として開発チーム規模、機能の豊富さ、Alphabet外での採用実績の多さが挙げられ、Haikuはバグ修正・JAX互換性維持中心のメンテナンスモードへ移行するとされている。","title":"google-deepmind/dm-haiku README（Flaxへの移行推奨に関する記述）","coverageType":["Differentiation"],"sourceType":"github","sourceClass":"Documentation","sourceUrl":"https://github.com/google-deepmind/dm-haiku","confidence":"high","supportedPromptTypes":["P-06"],"needsVerification":true,"sourceVerified":false,"sourceKind":"official","entityId":"flax"}],"generatedAt":"2026-08-29T14:34:38.294Z","evidenceIds":["flax-ev-cr2-deepmind-haiku-recommendation"]},{"id":"P-04-002","companyId":"flax","questionId":"P-04-002","instanceId":"reference-depth-completion-run-cohort2-unit-b","draftId":"reference-depth-completion-run-cohort2-unit-b-flax-p-04-002","promptText":"Flaxは実際にどのような大規模言語モデルの構築に使われていますか？","promptTypeId":"P-04","answer":"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が研究用途に留まらず実際の公開モデルファミリーの構築基盤として使われた具体例です。","evidencePoints":["flax-ev-cr2-gemma-recurrentgemma-usage"],"scope":"","differentiation":"","faq":[],"pageUrl":"https://www.refbase.ai/reference/flax/P-04-002","sourceEvidence":[{"id":"flax-ev-cr2-gemma-recurrentgemma-usage","text":"Google AI for Developersの公式ページに『Flax was used to create Gemma』と明記されており、RecurrentGemma（Griffinモデル）のライブラリもJAX・Flax・Orbax・SentencePieceで実装されている。GriffinモデルはFlaxのflax.linen.Moduleを継承する構造を持つ。","title":"Fine-tuning RecurrentGemma using JAX and Flax（Google AI for Developers）","coverageType":["Capability"],"sourceType":"product_docs","sourceClass":"Documentation","sourceUrl":"https://ai.google.dev/gemma/docs/recurrentgemma/recurrentgemma_jax_finetune","confidence":"high","supportedPromptTypes":["P-04"],"needsVerification":true,"sourceVerified":false,"sourceKind":"official","entityId":"flax"}],"generatedAt":"2026-08-29T14:44:28.527Z","evidenceIds":["flax-ev-cr2-gemma-recurrentgemma-usage"]}]}