Who contributed to Google on May 12, 2026?

1 developer shipped this update, including copybara-service[bot].

What were the notable Google updates?

[Mosaic GPU] Relax the divisibility constraint for loads/stores with gather indices, Fixes for JAX Bzlmod builds, and * Rename `auto_pvary` to `auto_pcast` since it disables more than just pvarys.

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 OVERHAULS GPU BUILD SYSTEM; GENAI SHIPS BLOCKING FUNCTION CALLS

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

1 person shipped this

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

JAX fixed critical Bzlmod build failures across Windows, CUDA, and ROCm while python-genai v2.1.0 lands server-side tool deltas and video response parameters.

The JAX team shipped a comprehensive fix [2] for Bzlmod builds that restores Windows and GPU compilation after curl patches became unnecessary and cross-compile toolchains stopped working with the new build system. Separately, they relaxed Mosaic GPU's overly strict divisibility constraints on gather index operations [1], unblocking production workloads that had been constrained by unnecessary alignment requirements. The team also renamed `auto_pvary` to `auto_pcast` [3] to better reflect its actual behavior and exposed the config flag, tidying up the public API. A critical Mosaic NCCL rollback [4] landed after breaking OSS tests, so if you're tracking main, revert and wait for the fixed version. Over in python-genai, v2.1.0 dropped with blocking FunctionCall support in Live API [5], server-side tool deltas [6], and gemini-3.1-flash-lite model support [7]. The google-cloud-python team simultaneously dropped Python 3.9 support [8] across api-core and removed it from default nox sessions [10], forcing upgrades for teams still on older runtimes. SQLAlchemy 2.0 compatibility fixes [11] and pandas test cleanup [9] round out the week's stability work.

Action items

→ Verify JAX Bzlmod Windows/CUDA/ROCm builds work in your CI — if you use GPU, test immediately google/jax [immediate]
→ Update python-genai to v2.1.0 if you use Live API with function calling googleapis/python-genai [plan]
→ Upgrade to Python 3.10+ for google-api-core; Python 3.9 is no longer tested googleapis/google-cloud-python [plan]
→ Monitor JAX main for Mosaic NCCL re-fix; current version has known regression google/jax [monitor]

References

[1] [Mosaic GPU] Relax the divisibility constraint for loads/stores with gather indices. ↗ google/jax
[2] Fixes for JAX Bzlmod builds ↗ google/jax
[3] * Rename `auto_pvary` to `auto_pcast` since it disables more than just pvarys. ↗ google/jax
[4] [Mosaic:GPU] Rollback: Use NCCL API on the host to collective peer addresses. ↗ google/jax
[5] feat: support Blocking FunctionCall in Live API in AgentPlatform (Vertex) ↗ googleapis/python-genai
[6] feat: Introduce Server Side tools deltas ↗ googleapis/python-genai
[7] v2.1.0 ↗ googleapis/python-genai
[8] fix(api-core): drop support for Python 3.9 (#16980) googleapis/google-cloud-python
[9] test: remove pandas version restrictions and fix precision issue (#17005) googleapis/google-cloud-python
[10] chore: remove Python 3.9 from default nox session (#17017) googleapis/google-cloud-python
[11] fix(sqlalchemy-bigquery): update literal binds test for SQLAlchemy 2.0 (#17029) googleapis/google-cloud-python

Quick answers

What shipped in Google on May 12, 2026?: JAX fixed critical Bzlmod build failures across Windows, CUDA, and ROCm while python-genai v2.1.0 lands server-side tool deltas and video response parameters. In total, 45 commits, 47 pull requests, and 2 releases landed.
Who contributed to Google on May 12, 2026?: 1 developer shipped this update, including copybara-service[bot].
What were the notable Google updates?: [Mosaic GPU] Relax the divisibility constraint for loads/stores with gather indices, Fixes for JAX Bzlmod builds, and * Rename `auto_pvary` to `auto_pcast` since it disables more than just pvarys.

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.