教程

• 即时编译
• 自动向量化
• 自动微分
• 调试简介
• 伪随机数
• 使用 Pytree
• 并行编程简介
• 有状态计算
• JIT 下的控制流和逻辑运算符

高级教程

• 高级自动微分
• 外部回调
• 使用 jax.checkpoint (jax.remat) 进行梯度检查点
• JAX 内部：原语
• JAX 内部：jaxpr 语言