jax.extend 模块#
用于 JAX 扩展的模块。
The jax.extend 包提供了访问 JAX 内部机制的模块。参见 JEP #15856。
API 政策#
与 公共API 不同，此包在版本之间 不提供兼容性保证。重大更改将通过 JAX 项目更新日志 宣布。
jax.extend 模块#用于 JAX 扩展的模块。
The jax.extend 包提供了访问 JAX 内部机制的模块。参见 JEP #15856。
与 公共API 不同，此包在版本之间 不提供兼容性保证。重大更改将通过 JAX 项目更新日志 宣布。