Who contributed to Google on May 5, 2026?

2 developers shipped this update, including copybara-service[bot] and noahdietz.

What were the notable Google updates?

Remove usages of xla_gpu_experimental_aot_compiled_thunks, Create `pl.select_ref` to dynamically choose from multiple refs for DMAs, and Add a jaxpr_html IR dump mode that dumps a jaxpr to an HTML document.

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 TIGHTENS GPU COMPILATION PIPELINE WHILE GENAI CLIENT GAINS FILE SEARCH

By RepoJournal · Filed 06:03 UTC on May 5, 2026 · About Google

2 people shipped this

copybara-service[bot] @copybara-service[bot] 4 cited

noahdietz @noahdietz 1 cited

JAX is consolidating its GPU optimization path by removing experimental flags and shipping dynamic ref selection for data movement, while the Python GenAI client reaches v1.75.0 with multimodal file search support.

JAX's GPU stack is maturing fast. The team removed the xla_gpu_experimental_aot_compiled_thunks flag [1], which is now enabled by default, signaling a move away from experimental compilation modes toward production-ready AOT pathways. Simultaneously, they shipped pl.select_ref [2] — a new primitive that lets you dynamically choose between multiple reference objects for DMAs, supporting complex nested transforms and async copies that were previously awkward to express. This is foundational for building more flexible data movement patterns on TPU. On the visualization front, jaxpr_html [3] gives you actual HTML dumps with full source tracebacks, making it orders of magnitude easier to debug complex computation graphs. Meanwhile, Mosaic GPU got two critical lowering improvements: maskTiledVregs consolidation [4] and n-d transpose support [5], both pushing the compiler closer to handling real workloads. Over in the Python ecosystem, python-genai v1.75.0 [6] landed with multimodal file search capabilities, fixing a stale token caching bug in async mTLS [6] that could bite production deployments. The google-cloud-python monorepo is dropping Python 3.7–3.9 support [7], aligning with end-of-life, and enabling gRPC transport for memorystore clients [8].

Action items

→ Review jaxpr_html dump mode for your complex models — it's production-ready and worth integrating into your debugging workflow google/jax [plan]
→ Update to python-genai v1.75.0 if you use async mTLS — the token caching fix is critical for production googleapis/python-genai [plan]
→ Audit your google-cloud-python usage — 3.10+ is now the floor; plan migrations off 3.9 and earlier googleapis/google-cloud-python [monitor]

References

[1] Remove usages of xla_gpu_experimental_aot_compiled_thunks google/jax
[2] Create `pl.select_ref` to dynamically choose from multiple refs for DMAs. ↗ google/jax
[3] Add a jaxpr_html IR dump mode that dumps a jaxpr to an HTML document. ↗ google/jax
[4] [Mosaic] Refactor maskTiledVregs and selectWithBounds to consolidate usages of mask and select. ↗ google/jax
[5] [Mosaic GPU] Lower n-d transposes in WG semantics. ↗ google/jax
[6] v1.75.0 ↗ googleapis/python-genai
[7] chore(versions): update Python version support to 3.10+ in gapic-generator (#16910) googleapis/google-cloud-python
[8] feat(memorystore): enable gRPC transport ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on May 5, 2026?: JAX is consolidating its GPU optimization path by removing experimental flags and shipping dynamic ref selection for data movement, while the Python GenAI client reaches v1.75.0 with multimodal file search support. In total, 31 commits, 29 pull requests, and 1 releases landed.
Who contributed to Google on May 5, 2026?: 2 developers shipped this update, including copybara-service[bot] and noahdietz.
What were the notable Google updates?: Remove usages of xla_gpu_experimental_aot_compiled_thunks, Create `pl.select_ref` to dynamically choose from multiple refs for DMAs, and Add a jaxpr_html IR dump mode that dumps a jaxpr to an HTML document.

JAX FIXES REMAT3 CONSTANT HANDLING CRASH, GOOGLE CLOUD PYTHON CENTRALIZES REST TRANSCODING

Matthew Johnson patched a critical remat3 regression where closed-over constants were silently dropped, causing immediate crashes in production rematerialization pipelines [ref:1].

ai 50 shipped 1-min read

Google 1 day ago

JAX TIGHTENS XLA OPTIMIZATION CONTROLS, GOOGLE CLOUD PYTHON SETS UP FEATURE GATES

JAX makes XLA optimization levels human-readable while Google Cloud Python client libraries prepare a unified feature gating system for observability.

ai 54 shipped 1-min read

Google 2 days ago

GENAI AUDIO TRANSCRIPTION SHIPS, PROTO-PLUS FIXES RACE CONDITIONS

Google's AI SDK gains native audio processing while the core Python cloud libraries ship critical stability fixes across the stack.

ai 60 shipped 1-min read

Google 3 days ago

JAX CUTS ASYNC COLLECTIVE PRIMITIVES, GENAI SDK ADDS LIVE API VOCABULARY

JAX is simplifying its async collective operations while the Python GenAI SDK ships custom vocabulary support for the Live API, marking a shift toward leaner abstractions and richer real-time capabilities.

ai 100 shipped 1-min read

Elsewhere on the wire

AI Agents about 9 hours ago

CLAUDE OPUS 5 LANDS ACROSS THE STACK

The newest Anthropic model is now live in langchain, Cline, and llama-index, with native support for extended reasoning and 1M context windows.

ai-agents 28 shipped 1-min read

Local LLMs about 9 hours ago

OLLAMA LANDS LAGUNA SUPPORT AND CRUSHES MEMORY LEAKS WHILE SGLANG HITS V0.5.16 WITH CONFIDENCE-DRIVEN SPECULATIVE DECODING

Ollama shipped three critical performance and reliability fixes for Metal residency and concurrent access patterns, while SGL-Lang released 0.5.16 with a new speculative algorithm hitting 383.7 tok/s on DeepSeek-V4.