编程

1 篇文章

2026年10月13日如何扩展你的模型（10）：用 JAX 给 TPU 编程中英对照第 10 篇：jax.jit / pmap / shard_map、GSPMD、pallas kernel 等并行 API 到底怎么用，配可运行的例子。左栏原文，右栏译文。JAX XLA TPU 编程中英对照