JAX
ソフトウェア
学習
エージェント
Composable function transforms — the framework Gemini is built with.
これはソフトウェアのレコードです。AIスタックを構成するツールまたはプラットフォームについて、開発・保守を担う主体と対応ハードウェアを紹介します。
写真: Google LLC · Public domain · Wikimedia Commons
JAX treats models as pure functions to be transformed — grad, jit, vmap, pmap — compiling through XLA to TPU pods with exceptional scaling ergonomics. It is Google DeepMind's house framework (Gemini training) and the favorite of much of the research frontier outside PyTorch.
スタック全体とのつながり
このレコードのリンク一覧 — グループをクリックするとジャンプします。
動作環境
メンテナンス担当
競合製品
出典
レコードのレビュー時点： 2026-01-15 — 表示 調査手法.