JAX 训练加速指南：8 个让 TPU 满跑的工程实战习惯
TPU 训练的真实效率往往取决于两个核心要素：**Shape 的稳定性**与**算子的融合度**。
JAX 核心特性详解：纯函数、JIT 编译、自动微分等十大必知概念
如果你用过 NumPy 或 PyTorch，但还没接触过 JAX，这篇文章能帮助你快速上手。
Google开源Tunix：JAX生态的LLM微调方案来了
**Tunix(Tune-in-JAX)**是一个**用于LLM后训练的JAX原生库**,旨在通过JAX的速度和可扩展性简化监督微调、强化学习和蒸馏，可以与Flax NNX无缝集成。
JAX快速上手：从NumPy到GPU加速的Python高性能计算库入门教程
JAX代表了Python高性能数值计算领域的重要进展。通过提供与NumPy兼容的接口，结合强大的函数转换能力以及基于加速线性代数的高效硬件执行
使用JAX实现完整的Vision Transformer
本文将展示如何使用JAX/Flax实现Vision Transformer (ViT)，以及如何使用JAX/Flax训练ViT。
