jax-ml 收录中的开源项目，按星标数排序。
可组合的 Python+NumPy 程序转换：微分、向量化、JIT 到 GPU/TPU 等功能
《How To Scale Your Model》的主页，这本短篇博客风格教材介绍了如何在 TPU 上扩展 LLM
