Google DeepMindが公式に保守するJAX向け別のニューラルネットワークライブラリ『Haiku』のGitHubリポジトリ（google-deepmind/dm-haiku）のREADMEには、『2023年7月時点でGoogle DeepMindは新規プロジェクトにHaikuではなくFlaxの採用を推奨する』と明記されています。理由として、Flaxの方が開発チームの規模が大きく、機能が豊富で、Alphabet外部での採用実績も多い点が挙げられており、Haiku自体は今後バグ修正とJAX互換性維持を中心とするメンテナンスモードに移行するとされています。この記述はFlax自身の公式ページではなく、Google DeepMindが別プロダクトの文脈で発した独立した言及である点に留意してください。