文章分类
Paper 30 OS 49 Code & Development 38 AI infra 12 DB 2 Algorithm 2
JAX 101（二）：函数变换、自动微分与批处理 JAX 101（二）：函数变换、自动微分与批处理
从纯函数开始介绍 JAX 的 grad、value_and_grad、vmap、jit 与 tracing，并通过一次梯度更新理解函数变换的组合方式。
2026-05-23 AI infra
JAX 自动微分 vmap JIT
JAX 101（一）：数组、jax.numpy 与语言模型的数据表示 JAX 101（一）：数组、jax.numpy 与语言模型的数据表示
从 JAX 数组、不可变更新、shape、dtype 和设备放置出发，理解 token、embedding 与语言模型 batch 在 JAX 中的数据表示方式。
2026-05-16 AI infra
JAX jax.numpy XLA 大语言模型
2 / 2