jax.enable_custom_vjp_by_custom_transpose

内容

jax.enable_custom_vjp_by_custom_transpose = <jax._src.config.State object>#

用于 jax_enable_custom_vjp_by_custom_transpose 配置选项（瞬态）的上下文管理器。

启用一个内部升级，该升级通过减少到 jax.custom_jvp 和 jax.custom_transpose 来实现 jax.custom_vjp。这将在未来版本的 JAX 中默认启用，届时所有对该标志的使用都将被视为已弃用（遵循 API 兼容性策略）。