jax-ml 收錄中的開源專案，依星數排序。
可組合的 Python+NumPy 程式轉換：微分、向量化、JIT 至 GPU/TPU 等功能
《How To Scale Your Model》的主頁，這本短篇部落格風格教材介紹了如何在 TPU 上擴充 LLM
