RepoJournal
Google

Google

JAX, the GenAI SDK, and the Cloud libs — Google's open source layer

Pick a date

Topics: AI / ML Python Full archive →

The Wire · Showcase

JAX BACKWARD PASS NOW LOGS DEBUG DATA. CLOUD AUTH TIGHTENS mTLS PRECEDENCE

By RepoJournal · Filed · About Google

JAX's autodiff pipeline shipped structured logging from transpose rules, letting you inspect what's happening inside the backward pass without tanking performance.

The backward pass logging feature [1] is the headline: transpose rules can now return named pytrees of data that flow out through `f_vjp.with_logs(out_ct)`, returning both arg cotangents and a logs dict. Plain `f_vjp(out_ct)` still works and drops the logs entirely, so JIT compiles them away. Existing rules need zero changes; they just keep returning None from the new output slot. This composes with gradient refs for even finer instrumentation. Alongside that, the vjp3 infrastructure caught up to recent AD features [2], with xla_call_metadata_call and compute_on rules now tracking lojax rules, structured residuals, and backward pass logging. The docs also got a refresh [3] on custom derivatives with clearer rules sections and new examples. On the auth front, google-cloud-python consolidated its mTLS cert discovery logic [4] into a unified helper, killing scattered checks across multiple functions. More importantly, X.509 Workload Identity Federation now takes hard precedence over SecureConnect by checking the consolidated path first, tightening the precedence order.

Action items

References

  1. [1] [vjp3] add a way to log data out of the backward pass ↗ google/jax
  2. [2] [vjp3] xla_call_metadata_call / compute_on lojax rules, sres, logging ↗ google/jax
  3. [3] [docs] improve 301 page on custom derivatives with hijax ↗ google/jax
  4. [4] fix(auth): centralize cert discovery logic and steps ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on July 19, 2026?
JAX's autodiff pipeline shipped structured logging from transpose rules, letting you inspect what's happening inside the backward pass without tanking performance. In total, 10 commits and 5 pull requests landed.
Who contributed to Google on July 19, 2026?
2 developers shipped this update, including mattjj and nbayati.
What were the notable Google updates?
[vjp3] add a way to log data out of the backward pass, [vjp3] xla_call_metadata_call / compute_on lojax rules, sres, logging, and [docs] improve 301 page on custom derivatives with hijax.

More from Google

Daily updates, in your inbox

Follow Google

JAX, the GenAI SDK, and the Cloud libs — Google's open source layer We'll email you a link to confirm first.

Free. Confirm via email. Unsubscribe in one click.

— or follow the whole beat:

Elsewhere on the wire