TensorTonic 推出 JAX for ML Systems 学习计划,35 道手写题覆盖 KV 缓存到并行训练

burny_tech · x · 2026-11-10

ML 学习平台 TensorTonic 上线了新的 JAX 学习计划「JAX for ML Systems」,围绕 ML 系统底层构建,共 35 道编程题(10 简单 / 16 中等 / 9 困难),分 9 个章节。

内容覆盖:用 vmap 计算逐样本梯度、混合精度训练与 loss scaling、构建分页 KV 缓存、speculative decoding、函数式数组与 PRNG key、jit/pytree 优化器到并行训练等。定位是「PyTorch 教你训练模型,JAX 教你理解底层系统」,适合想深入推理/训练系统工程细节的工程师。

原文链接 →

「编程与Agent」频道最新

更多「编程与Agent」频道 AI 资讯 →