Jax的學習資源 (編輯中)

Jax作為較年輕且擁有較高靈活性的高效能數值計算庫,不僅擁有較陡峭的學習曲線,學習資源也相對零碎。這篇主要用來整理我認為網路上比較值得拿來參考的Jax學習資源

官方文檔

官方文檔毫無疑問是最直接的學習資源
https://jax.readthedocs.io/en/latest/

線上教科書

Dive into Deep Learning

Deep Learning的聖經之一,雖然主要在介紹Deep Learning,但部分章節有提供Jax的程式碼範例,可以在學習Deep Learning的同時學習如何編寫Jax的代碼
https://d2l.ai/index.html

Quantitative Economics with JAX

在學習經濟學的同時,學習如何使用Jax來解決數學問題

Youtube

這是我認為Youtube上最值得參考的系列教學之一,一共四個影片

Hugging Face的教學,我想應該也很值得參考

這個系列比較麻煩的是第一個影片設為私人了,但wandb的論壇上可以找到這系列教學的colab notebook連結。播放清單的前面幾部影片算是入門教學,後面的影片看起來像是Jax生態系的一些相關套件的介紹(?

1 Like