JAX v0.11.0

What moved

New features Added a doc on defining custom derivative rules with the experimental hijax API ( hijax-custom-derivatives ), along with jax.experimental.hijax helpers for deriving VJPHiPrimitive autodiff rules from a jvp or lin rule: linearize_from_jvp with apply_derived_linearization , vjp_fwd_from_jvp with transpose_jvp , vjp_fwd_from_lin with transpose_linearized , and jvp_from_lin . Added jax.custom_remat to the top-level jax namespace, for per-function control of rematerialization under the new jax_remat3 implementation. jax.checkpoint_policies is now a submodule rather than a namespace object (so from jax.checkpoint_policies import ...

Why it matters

New features Added a doc on defining custom derivative rules with the experimental hijax API ( hijax-custom-derivatives ), along with jax.experimental.hijax helpers for deriving… That is a public note from JAX, dated 2026-07-16. Tagged Hardware.

On the record

• Filed from the JAX official RSS on 2026-07-16.
• Primary source host: github.com.
• The desk does not add a second source.