jax.example_libraries 模块#

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

注意

每个小型库的目的是作为一种启发，而不是一种规范。

为了达到这个目的，最好保持代码示例的简洁；因此，我们通常不会合并添加新功能的 PR。相反，请将您精彩的 Pull Request 和设计理念发送给功能更完善的库，例如 Haiku 或 Flax。