教學文件

• 快速入門
• 重要概念
• 即時編譯
• 自動向量化
• 自動微分
• 除錯簡介
• 虛擬隨機數
• 使用 pytrees
• 平行程式設計簡介
• 具狀態的計算
• 使用 JIT 的控制流程和邏輯運算子

進階教學文件

• 進階自動微分
• 外部回呼
• 使用 jax.checkpoint (jax.remat) 的梯度檢查點
• JAX 內部機制：基本運算
• JAX 內部機制：jaxpr 語言