jax.extend 模組#
用於 JAX 擴充功能的模組。
jax.extend 模組提供用於存取 JAX 內部機制的模組。請參閱 JEP #15856。
此模組並非 JAX 旨在實現可擴充性的唯一方法。例如，主要的 JAX API 提供了用於自訂導數、註冊自訂 pytree 定義等的機制。
API 政策#
與公開 API 不同，此模組不提供跨版本發布的相容性保證。重大變更將透過 JAX 專案變更日誌 公告。
jax.extend 模組#用於 JAX 擴充功能的模組。
jax.extend 模組提供用於存取 JAX 內部機制的模組。請參閱 JEP #15856。
此模組並非 JAX 旨在實現可擴充性的唯一方法。例如，主要的 JAX API 提供了用於自訂導數、註冊自訂 pytree 定義等的機制。
與公開 API 不同，此模組不提供跨版本發布的相容性保證。重大變更將透過 JAX 專案變更日誌 公告。