JAX 生态系统简介

至此，尽管 JAX 拥有丰富的功能，但在课程进行到目前阶段时，它似乎并不像一个专门的神经网络库。这可以理解，因为 JAX 的核心本质上是一个可并行化的 NumPy，并附带了许多其他功能。

然而，故事并未就此结束。为了支持深度学习，JAX 拥有自己的一套生态系统，包含以下主要库：

1. Haiku
2. Optax
3. Jraph
4. RLax
5. Chex

这些库中的每一个都值得单独用一章来详细介绍，即便如此也难以穷尽其全部细节。但在本入门课程中，我们仅对每个库做简要概述。

注意： Flax 是一个神经网络库，但通常不被视为 JAX 生态系统的一部分。