鐵人檔案
TensorFlow 和 Pytorch 是現今學界及業界使用最頻繁的 AI 框架。從數據資料的處理，模型的建構及訓練，到模型的布建及應用，它都提供了相當完備的支援。相較於 TensorFlow 和 Pytorch，2019 年發表的 JAX，似乎開闢了另外一條人工智慧框架之路。
「JAX 好好玩」這一系列的發文，希望能夠由淺入深，把 JAX 好好玩一遍。藉由實際的操作及練習，輔以文字的說明，讓讀者能夠快速的熟悉 JAX，知道它的優點和它不足的地方；也讓讀者能夠在自己的專案上適當的應用 JAX 提供的各種功能，而不致於對 JAX 有錯誤的期待。
這張提到 JAX 的投影片來自 State of AI Report 2021 (https://www.stateof.ai/2021) 這份報告的第 74...
我想絕大多數的 AI 人，都很熟悉 Colab 的使用方法，如果你從來沒有使用過 Colab，不妨點擊這個網址開始學習 (colab 介紹)。 老頭個人非常喜歡...
在正式進入 JAX 之前，老頭先整理了 JAX 學習資源，分享給大家。如果你有其他的好東西，請你也分享給老頭。 1. JAX reference documen...
還記得老頭最初接觸 JAX 的前一個月，雖然從各類的文獻中讀到有關 JAX 的定義、它的組成元件、和它的某些應用，但總是摸不到對於 JAX 的「感覺」 ，NO...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，可由此下載） 在 JAX 官方的教學網頁 JAX Quickstart [5.1] 上，開宗明義就說：...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，可由此下載） 雖然 jax.numpy 是依據 Numpy 的語法和語義來設計的，但仍有幾個不同點需要...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，可由此下載） 上回老頭介紹了 JAX PRNG 的基本使用方法，現在讓我們更深一層的來探討它。 JAX...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，可由此下載） 極速實驗的結果 現在我們要來看看 jax.numpy 執行起來比傳統的 Numpy 快多...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，可由此下載） 之前曾經提到，DeviceArray 是 JAX 自行定義的陣列類別，定義在 jax.n...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，可由此下載） 我們更進一步的來研究當 DeviceArray 陣列索引 (index) 超過範圍 (o...
