JAX 教學文件草稿# 注意 以下教學文件為草稿;目前,請參考較舊的教學內容,包括JAX 入門指南、使用者指南,以及現已刪除的JAX 101 教學文件。 JAX 101# 大致已在教學文件中完成! 快速入門 重要概念 即時編譯 自動向量化 自動微分 除錯簡介 虛擬隨機數 使用 pytree 平行程式設計簡介 具狀態的計算 範例:撰寫簡單神經網路 JAX 201# 平行計算 進階除錯 效能分析與效能 JAX 301# 進階編譯