jax.example_libraries module#

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

注意

每个小型库都旨在提供灵感，而非规定。

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