jax.example_libraries 模块#

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

注意

每个迷你库都旨在作为灵感，而不是规范。

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