{"ok":true,"entity":{"slug":"flax","entityType":"product","name":"Flax","officialName":"Flax","canonicalName":"Flax","displayName":"Flax","category":"JAX向けニューラルネットワークライブラリ","shortDescription":"Google DeepMindのチームが維持する、柔軟性を重視したJAX向けニューラルネットワークライブラリ。","alias":[],"searchKeywords":["JAX","neural network library"],"website":"https://github.com/google/flax","parentEntity":"google-deepmind","primaryCluster":"ai-infrastructure","secondaryClusters":[],"verificationStatus":"draft","id":"flax","updatedAt":"2026-07-20T08:41:20.341Z"},"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"}]}