从 @jax.jit 到 VLIW:追踪 TPU 编译器的每一层2026年4月6日·14 分钟TPU Pallas TPU JAX编译器通过一行@jax.jit 将8行代码就优化到了250 VLIW bundles. 本文通过HLO/LLO追踪了 TPU 编译器的优化之旅