jax.example_libraries 模块

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

备注

每个迷你库旨在成为一个 灵感，但不是一个处方。

为了达到这个目的，最好保持他们的代码示例简洁；因此我们通常 不会合并添加新功能的PR。相反，请将您的优秀拉取请求和设计想法发送到功能更全面的库，如 Haiku 或 Flax。

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