RepoJournal
Google Google
140 wires and counting

$ follow Google

Keep up with Google in about 3 minutes: what actually shipped — the commits, pull requests, releases, and security advisories that matter.

or

fair warning: these emails are deeply technical. diffs, version numbers, CVEs, benchmark deltas. if that's not your idea of a good read, this isn't your newsletter.

Folds into your digest — weekly by default, monthly if you prefer. Unsubscribe in one click.

$ status

wire 2026-09-12
stories 51
RSS Archive Reviews

© 2026 RepoJournal Home Showcase How it works Privacy

$ the-wire · showcase

JAX pins down DCE jaxpr identity, fixes i0/i1 gradients at zero

By RepoJournal · Filed · About Google · Composed from the cited sources · methodology

copybara-service kodlan jakevdp chalmerlowe suztomo mattjj Wanlin Du
10 people shipped this
copybara-service @copybara-service 4 cited kodlan @kodlan 1 cited jakevdp @jakevdp 1 cited chalmerlowe @chalmerlowe 1 cited suztomo @suztomo 1 cited mattjj @mattjj Wanlin Du @wanlin31 Jaycee Li @jaycee-li ohmayr @ohmayr Min Zhu @zhumin8

JAX fixed a nondeterminism bug where lowering the same jitted function could emit different StableHLO depending on what the process had traced earlier, and corrected higher-order gradients of i0 and i1 at 0.0.

Lowering the same jitted function used to emit different StableHLO depending on trace history. The MLIR lowering cache dedups repeated subcomputations by jaxpr object identity, but DCE always rebuilt a new jaxpr even when it eliminated nothing, so that identity was only stable while the right weakref LRU cache entries stayed warm. The fix makes no-op DCE return the input jaxpr [1], and the PR notes it fixes the reported repro while one narrower subclass remains open.

The JVP rule for `jax.numpy.i0` used to return a constant zero at `x=0.0`, producing incorrect higher-order derivatives; it now uses a maclaurin series approach for the JVP of i1, similar to the approach used for `sinc` [2]. JAX also added a `lax.one_minus_square` primitive to preserve accuracy of `1 - x^2` when evaluating near `+/-1` and differentiating near `0`, addressing the tension in expressions such as the derivatives of `tanh`, `atanh`, `asin`, and `acos` [3]. Separately, Ref buffers now keep their memory space constraints during state discharge [4].

In python-genai, the automatic function calling loop no longer runs functions once its budget is spent [6], closing a path where calls continued past the configured limit. RetrievalCallStep and RetrievalResultStep were added to the interactions schema and SDKs [5].

google-cloud-python landed Tier 3 client method span wrapping in `google.api_core.gapic_v1.method._GapicCallable.__call__`, starting an OpenTelemetry `SpanKind.CLIENT` span around the high-level GAPIC SDK method call and setting `rpc.system = "grpc"`, `rpc.service`, and `rpc.method`, encompassing client preparation, retry loops, timeouts, and error handling [7]. `google-cloud-spanner` moves to its own release PR for independent release management [8].

Action items

References

  1. [1] Make no-op DCE return the input jaxpr, so lowering does not depend on trace history ↗ google/jax
  2. [2] [autodiff] fix first, second, and higher-order gradients of i0 and i1 at 0.0 ↗ google/jax
  3. [3] Add `lax.one_minus_square` primitive to preserve accuracy of `1 - x^2` when evaluating near `+/-1` and differentiating near `0`. ↗ google/jax
  4. [4] Preserve memory space constraints on Ref buffers during state discharge. ↗ google/jax
  5. [5] feat: Add RetrievalCallStep and RetrievalResultStep to interactions schema and SDKs. ↗ googleapis/python-genai
  6. [6] fix: do not run functions once the automatic function calling budget is spent ↗ googleapis/python-genai
  7. [7] feat(gapic): add OpenTelemetry T3 client method span wrapping in gapic_v1.method (D) ↗ googleapis/google-cloud-python
  8. [8] chore(spanner): move to individual releases ↗ googleapis/google-cloud-python

$ ls google/reviews/ # the step-back reads

the week in review JAX drops legacy IFRT device APIs, google-auth adds 3.15 the month in review JAX free-threading push reshapes 0.11.2 and 0.12 cleanup
Full archive → Weekly reviews →

Quick answers

What shipped in Google on September 12, 2026?
JAX fixed a nondeterminism bug where lowering the same jitted function could emit different StableHLO depending on what the process had traced earlier, and corrected higher-order gradients of i0 and i1 at 0.0. In total, 28 commits and 23 pull requests landed.
Who contributed to Google on September 12, 2026?
10 developers shipped this update, including kodlan, jakevdp, copybara-service, mattjj, Wanlin Du, Jaycee Li, chalmerlowe, and ohmayr, and 2 more.
What were the notable Google updates?
Make no-op DCE return the input jaxpr, so lowering does not depend on trace history, [autodiff] fix first, second, and higher-order gradients of i0 and i1 at 0.0, and Add `lax.one_minus_square` primitive to preserve accuracy of `1 - x^2` when evaluating near `+/-1` and differentiating near `0`.
♥

$ log Google --recent

JAX pins OpenXLA revision, disables overlap_test against older libtpu 5 shipped JAX adds type-2 and type-3 Hermitian eigensolves, lexsort batching cuts unique compile time 69 shipped JAX hijax flags read at call time, Spanner stops deleting multiplexed sessions 112 shipped Vertex AI client timeout fix, Compute v1 API regenerated 87 shipped

$ log --all # elsewhere on the wire

@vuejs Vue language tools stop calling registerHooks on buggy Node builds @openclaw OpenClaw's deslop sweep removes 1,000+ production lines

$ ls showcases/

Rails Django Laravel Node.js Go Vercel Vue.js Kubernetes HashiCorp Arch Linux Linux Anthropic

$ init your-desk

RepoJournal can write one of these for any repo you care about. Sign in with GitHub, pick your repos and sources, and your desk gets a wire like this one every morning.