JAX 教程草稿

注意

下面的教程仍在进行中；目前，请参阅旧的教程内容，包括 JAX 入门、JAX 101，以及已删除的《JAX 101》教程。

JAX 101

大部分已在 JAX 101 中定稿！

• 关键概念
• 即时编译
• 自动向量化
• 自动微分
• 调试简介
• 伪随机数
• 使用 Pytree
• 并行编程简介
• 有状态计算
• 示例：编写一个简单的神经网络

JAX 201

• 并行计算
• 高级调试
• 性能分析与优化

JAX 301

• 高级编译