趋近智
结业证书
JAX 入门授予
Hunaid Hameed
证书编号
xgkn-6dvw-puyr-vjgx
颁发日期
March 26, 2026
课程描述
学习 JAX，用于高性能数值计算和机器学习研究。本课程涵盖 JAX 核心知识，包括其 NumPy 接口、`jit`、`grad`、`vmap` 和 `pmap` 等函数变换，以及用于状态管理的函数式编程模式。您将获得适配现代硬件（GPU/TPU）来加速和求导 Python 代码的实践经验。
证书详情
课程难度:
专业级
颁发日期:
March 26, 2026
证书编号:
xgkn-6dvw-puyr-vjgx
学习成果
- 理解 JAX 的核心知识、它与 NumPy 的关联及其函数式编程方法。
- 应用 JAX 的主要变换：`jit` 用于编译，`grad` 用于自动求导，`vmap` 用于向量化，`pmap` 用于并行化。
- 编写能高效运用 GPU 和 TPU 等现代加速器的 JAX 代码。
- 使用 `grad` 自动计算 Python 函数的梯度。
- 使用适合 JAX 的函数式编程模式实现有状态计算。
- 识别调试 JAX 代码时遇到的常见问题和基本方法。