JAX 101# 这些教程涵盖了 JAX 及其功能的基本用法，包括使 JAX 工作的内部机制。它们从基础知识开始，应按顺序阅读。有关 JAX 设计和实现的更深入讨论，请按任意顺序参阅 进阶指南。 即时编译 自动向量化 自动微分 使用 Pytree 伪随机数 并行编程简介 JIT 下的控制流和逻辑运算符 追踪 有状态计算
