Google AI：DEV 作者专属· xbill·· 2026-08-29AI 评分56
Gemma 4 纯 JAX 移植实测：哪些能从 TPU 搬到 GPU，哪些不能
Gemma 4 in Pure JAX: What Ports from TPU to GPU, and What Doesn't
AI 导读
作者把一份手写的 Gemma 4 纯 JAX 端口跑在 Cloud TPU v5e、v6e 和 NVIDIA T4G 上，模型代码、编译缓存和静态形状无需改动即可跨这三种加速器复用。
来源：Google AI：DEV 作者专属 · dev.to