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 — 查看 方法论.