jax.example_libraries 模块#

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

注意

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

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