jax.example_libraries 模块

JAX 提供了一些用于机器学习的小型实验库。这些库部分是为了提供工具，部分是为了作为使用 JAX 构建此类库的示例。每个库的源代码行数都小于 300 行，所以请查看内部并根据需要进行调整！

注意

每个小型库都旨在成为一种启发，而不是一种规定。

为了达到这个目的，最好使它们的代码示例保持最小化；因此，我们通常不会合并添加新功能的 PR。相反，请将您优秀的拉取请求和设计思路发送到更完整的库，如 Haiku 或 Flax。

• jax.example_libraries.optimizers 模块
• jax.example_libraries.stax 模块