使用者指南

使用者指南更深入地探討 JAX 內部的特定主題，這些主題會隨著您的 JAX 專案發展成更大或已部署的程式碼庫而變得相關。

除錯與效能

• 如何以 JAX 思考
• 計算效能分析
• 裝置記憶體效能分析
• 除錯執行時期數值
• GPU 效能提示
• 持續性編譯快取

介面

• Pytrees
• 錯誤
• 預先降低和編譯
• 匯出和序列化
• 型別提升語意
• 傳輸保護

自訂操作

• Pallas：JAX 核心語言
• 外部函數介面 (FFI)

範例應用

• 訓練簡單神經網路，使用 tensorflow/datasets 資料載入
• 訓練簡單神經網路，使用 PyTorch 資料載入
• 貝氏推論的自動批次處理