Who contributed to Google on May 29, 2026?

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

What were the notable Google updates?

Error out if `*pspec` or `pspec[...]` is used when unreduced/reduced is non-empty, Add out_sharding to segment_{sum,max,min,prod}. Only allow unreduced out_sharding to `segment_sum`, and [Pallas] Improve error handling in TPU Interpret Mode to avoid hangs.

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 SEGMENT SHARDING RULES WHILE PALLAS FIXES CRITICAL TPU HANGS

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

1 person shipped this

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

JAX is cracking down on conflicting sharding specifications in segment operations while simultaneously shipping a critical fix that prevents TPU Interpret Mode from deadlocking when simulated devices fail.

The JAX team shipped two major enforcement changes to segment operations that will break existing code if you're mixing unreduced and reduced sharding directives [1] [2]. Segment sum, max, min, and prod now validate that you're not simultaneously using both parameter specs and unreduced/reduced output sharding, a pattern that was silently producing incorrect results. This is a hard requirement before the next major version lands. More critically, Pallas TPU Interpret Mode now properly handles device failures [3], implementing failure tracking and propagation so that when one simulated device crashes, others fail cleanly instead of hanging forever waiting for signals that will never arrive. The fix includes test cases that deadlock without it. Over in python-genai, the team shipped v2.7.0 [4] with computer_use field support for Vertex and CSV document mime type support for the Interaction API [5]. On the infrastructure side, google-cloud-python is consolidating CI/CD by moving prerelease and core_deps tests from GitHub Actions to Kokoro [6] [7], removing hardcoded package lists that were causing test coverage gaps [8].

Action items

→ Audit your JAX segment_sum/max/min/prod calls for conflicting sharding specs before next release google/jax [plan]
→ Upgrade python-genai to v2.7.0 if using Vertex computer_use or CSV document processing googleapis/python-genai [plan]
→ Update local Kokoro configuration references if you maintain google-cloud-python forks googleapis/google-cloud-python [monitor]

References

[1] Error out if `*pspec` or `pspec[...]` is used when unreduced/reduced is non-empty ↗ google/jax
[2] Add out_sharding to segment_{sum,max,min,prod}. Only allow unreduced out_sharding to `segment_sum`. ↗ google/jax
[3] [Pallas] Improve error handling in TPU Interpret Mode to avoid hangs. ↗ google/jax
[4] v2.7.0 ↗ googleapis/python-genai
[5] docs: update the docs for 2.7 ↗ googleapis/python-genai
[6] chore: update kokoro prerelease and core_deps jobs (#17286) googleapis/google-cloud-python
[7] chore: remove prerelease and core_deps from GHA (#17291) googleapis/google-cloud-python
[8] tests: remove hardcoded packages in system.sh (#17301) googleapis/google-cloud-python

Quick answers

What shipped in Google on May 29, 2026?: JAX is cracking down on conflicting sharding specifications in segment operations while simultaneously shipping a critical fix that prevents TPU Interpret Mode from deadlocking when simulated devices fail. In total, 29 commits, 25 pull requests, and 3 releases landed.
Who contributed to Google on May 29, 2026?: 1 developer shipped this update, including copybara-service[bot].
What were the notable Google updates?: Error out if `*pspec` or `pspec[...]` is used when unreduced/reduced is non-empty, Add out_sharding to segment_{sum,max,min,prod}. Only allow unreduced out_sharding to `segment_sum`, and [Pallas] Improve error handling in TPU Interpret Mode to avoid hangs.

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.