JAX 101

这些教程涵盖了 JAX 的基本用法及其功能，包括一些使 JAX 能够运行的内部机制。它们从基础知识开始，建议按顺序阅读。如需深入探讨 JAX 的设计与实现，请参阅进阶指南（阅读顺序不限）。

• 即时编译
• 自动向量化
• 自动微分
• Pytrees
• 伪随机数
• 分布式数组和自动并行化
• JIT 下的控制流和逻辑运算符
• 追踪
• 有状态计算