Module-scope @functools.partial(jax.jit, static_argnums=...) escapes M5. Match partial(jax.jit, ...) in decorator position.
From the pre-release multi-perspective review (2026-07-03). Follow CONTRIBUTING.md; every change ships a true-positive test + a false-positive guard test.
Module-scope @functools.partial(jax.jit, static_argnums=...) escapes M5. Match partial(jax.jit, ...) in decorator position.
From the pre-release multi-perspective review (2026-07-03). Follow CONTRIBUTING.md; every change ships a true-positive test + a false-positive guard test.