鐵人檔案
TensorFlow 和 Pytorch 是現今學界及業界使用最頻繁的 AI 框架。從數據資料的處理，模型的建構及訓練，到模型的布建及應用，它都提供了相當完備的支援。相較於 TensorFlow 和 Pytorch，2019 年發表的 JAX，似乎開闢了另外一條人工智慧框架之路。
「JAX 好好玩」這一系列的發文，希望能夠由淺入深，把 JAX 好好玩一遍。藉由實際的操作及練習，輔以文字的說明，讓讀者能夠快速的熟悉 JAX，知道它的優點和它不足的地方；也讓讀者能夠在自己的專案上適當的應用 JAX 提供的各種功能，而不致於對 JAX 有錯誤的期待。
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，可由此下載） JAX (jax.numpy) 在設計的時候，就儘量依循 Numpy 的語法和語義，但是...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，可由此下載 之前提到過的 Google Brain 2018 年的論文 Compiling machi...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，可由此下載 JAX JIT 使用原則之一：要被 jax.jit() 編譯的函式，應該要是純函式 (pu...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，可由此下載 我們現在來探討另外一個 JAX JIT 使用上的限制 [14.1]：函式運算的過程中，所有...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，可由此下載 JIT … work by tracing a function to determin...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，可由此下載 上一次的貼文提到以上這個例子，它在執行時會報錯。 @jax.jit def f2(x, n...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，可由此下載 老頭在之前「純函式 (Pure Function)」的貼文中，曾經舉了兩個例子，說明在 J...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，可由此下載 是時候來總結一下 JAX JIT 了 [18.1]。 追踪的過程 JAX JIT 的運作...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，[可由此下載](https://colab.research.google.com/drive/1tI...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，[可由此下載](https://colab.research.google.com/drive/1tI...
