RepoJournal
Google

Google

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

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

One email a day. Unsubscribe in one click.

Pick a date

Topics: AI / ML Python Full archive →

The Wire · Showcase

JAX 0.11.1 lands with export guardrails and a tracing speedup

By RepoJournal · Filed · About Google

JAX's latest release locks the door on expired exports while a core tracing hot path gets a serious speed boost.

JAX v0.11.1 [1] adds a hard error check for deserializing exports older than the compatibility window, closing a class of obscure downstream failures, and introduces a bypass flag for teams that need breathing room. The same release brings jax.numpy.top_k, but the real headline for heavy users is PR #40041 [2], which eliminates quadratic tracing behavior by no longer walking all previously traced equations when building error messages. That fix matters for frameworks like Flax NNX that probe Tracer.sharding and Tracer.device on every operation. Meanwhile, core_map_p is gone [3], with pl.core_map now implemented via mpmd_map_p, and the Pallas team removed MemoryRefBase [4] and transform_ref [5] outright, noting "I decided that a deprecation cycle is not necessary" for the latter. Over on the Python GenAI client, the live connection status enum gains an IDLE state while REQUIRES_ACTION gets deprecated [6]. Cloud Python users should note the bigtable data client now fails V3 mutate_rows entries the server never acknowledged [7], a correctness fix for batch writes, alongside an api_core rest path validation improvement [8].

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] JAX v0.11.1 ↗ google/jax
  2. [2] PR #40041: Avoid quadratic tracing behaviour google/jax
  3. [3] Removed `core_map_p` and everything which depended on it google/jax
  4. [4] [pallas] Removed `MemoryRefBase` google/jax
  5. [5] [pallas:mgpu] Removed `transform_ref` google/jax
  6. [6] feat: Add IDLE state to live connection status enum and mark REQUIRES_ACTION as deprecated. googleapis/python-genai
  7. [7] fix(bigtable): data client should acknowledge all mutations in batch ↗ googleapis/google-cloud-python
  8. [8] fix(api_core): improve rest path validation ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on August 18, 2026?
JAX's latest release locks the door on expired exports while a core tracing hot path gets a serious speed boost. In total, 35 commits, 27 pull requests, and 1 releases landed.
Who contributed to Google on August 18, 2026?
7 developers shipped this update, including hawkinsp, Sergei Lebedev, Olli Lupton, Mark Daoust, copybara-service, daniel-sanche, and mutianf.
What were the notable Google updates?
JAX v0.11.1, PR #40041: Avoid quadratic tracing behaviour, and Removed `core_map_p` and everything which depended on it.

More from Google

Daily updates, in your inbox

Follow Google

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

or

One email a day. Unsubscribe in one click. Read a past issue →

Elsewhere on the wire

Want every project, not just this one?