从零构建自动微分引擎:对偶数、计算图与反向传播的 Rust 深度实战 从零实现一个完整的自动微分引擎,深入理解对偶数、计算图构建、反向模式微分以及高阶导数的实现原理,用 Rust 实战演示 XOR 网络训练。 深度学习 2026年10月02日 0 点赞 0 评论 51 浏览
JAX/XLA 编译器基础设施深度解析:从 Python 函数到分布式 GPU 代码的编译之旅 深入拆解 JAX 的编译流水线,从 trace 机制到 HLO 优化、从自动微分规则推导到 GSPMD 分区器,揭示这套系统如何将 Python 代码转化为跨数百个 TPU/GPU 运行的高性能计算图。 编译原理 2026年10月04日 0 点赞 0 评论 58 浏览
Inside JAX:XLA 编译器如何将 Python 函数编译为 GPU PTX 代码 深入拆解 JAX/XLA 核心编译机制:Tracing 编译模型、HLO 中间表示、算子融合、自动并行化(GSPMD)、自定义VJP与性能调优实战 编译原理 2026年10月05日 0 点赞 0 评论 42 浏览