JAX 可微分编程框架
JAX 用 XLA 编译 NumPy 代码在 GPU/TPU 上加速运行。grad 自动求导jit 编译加速。vmap 向量化。本文用 JAX 训练线性回归模型。
14
0
0
2026-07-11
XLA 加速线性代数
学习 XLA 如何将计算图编译为优化的 HLO 内核，通过算子融合和内存优化加速 TensorFlow 和 JAX 中的线性代数运算。
20
0
0
2026-06-21