鐵人檔案
TensorFlow 和 Pytorch 是現今學界及業界使用最頻繁的 AI 框架。從數據資料的處理，模型的建構及訓練，到模型的布建及應用，它都提供了相當完備的支援。相較於 TensorFlow 和 Pytorch，2019 年發表的 JAX，似乎開闢了另外一條人工智慧框架之路。
「JAX 好好玩」這一系列的發文，希望能夠由淺入深，把 JAX 好好玩一遍。藉由實際的操作及練習，輔以文字的說明，讓讀者能夠快速的熟悉 JAX，知道它的優點和它不足的地方；也讓讀者能夠在自己的專案上適當的應用 JAX 提供的各種功能，而不致於對 JAX 有錯誤的期待。
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，[可由此下載](https://colab.research.google.com/drive/1tI...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，[可由此下載](https://colab.research.google.com/drive/1tI...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，[可由此下載](https://colab.research.google.com/drive/1tI...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，[可由此下載](https://colab.research.google.com/drive/1tI...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，[可由此下載](https://colab.research.google.com/drive/1tI...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，[可由此下載](https://colab.research.google.com/drive/1g5...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，[可由此下載](https://colab.research.google.com/drive/1g5...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，[可由此下載](https://colab.research.google.com/drive/1md...
（本貼文所列出的程式碼，皆以 colab 筆記本方式執行，可由此下載 簡介 Pytree 是 JAX 定義的資料結構，依照 JAX 官方文件 [29.1] ：...
在目前為止，老頭介紹了 jax.numpy、jax.jit、jax.grad、jax.vmap、Pytree 和 JAX 控制流程，以這些為基礎，我們可以試著設...
