Google Research 的 JAX 训练库 Kauldron 上手教程
这篇教程演示了 Google Research 的 JAX 训练库 Kauldron 的三个核心机制：konfig 把实验变成可 JSON 往返的普通字典树，kontext 用字符串键路径连接组件（损失函数无需 import 模型），以及运行时形状检查器。教程用合成内存数据训练了一个真实 Trainer，无需下载数据或加速器，还写了自定义 loss 和 metric，并演示了五组只差一行配置的 sweep 实验和断点续训。文中提到需为 jax 0.10.1 与 etils 的兼容问题打两行补丁。
推荐理由想用 JAX 做研究的人可以照着跑通，含 jax 与 etils 兼容补丁的具体代码