jax.example_libraries 模組#

JAX 提供一些用於機器學習的小型實驗性函式庫。這些函式庫一部分是為了提供工具，一部分是作為範例，說明如何使用 JAX 建構此類函式庫。每個函式庫都只有少於 300 行的原始碼，因此請查看內部並根據您的需要進行調整！

注意

每個迷你函式庫都旨在成為一種啟發，而不是一種規定。

為了達到這個目的，最好保持它們的程式碼範例盡可能簡潔；因此我們通常不會合併 PR 來新增功能。相反地，請將您優美的 pull request 和設計想法發送到功能更完整的函式庫，例如 Haiku 或 Flax。