RepoJournal
Google

Google

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

What actually shipped in Google, written up every day — commits, pull requests, releases, and security advisories.

One email a day. Unsubscribe in one click.

Pick a date

Topics: AI / ML Python Full archive →

The Wire · Showcase

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

By RepoJournal · Filed · About Google

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.

The JAX team is pruning its async collective primitive surface, removing redundant operations and committing to add only what's necessary [1]. This consolidation mirrors broader work to optimize test infrastructure: parameterizing sampler tests by JIT compilation status [2] cuts expensive recompilation overhead during subsampled test runs, a concrete win for developer velocity. Meanwhile, python-genai shipped v2.13.0 [3] with custom vocabulary support for Live API interactions and a new model selector, though deprecation warnings are now firing for Imagen's generate_images and edit_images methods when using legacy prompt/text/image arguments [4], signaling a major version break coming next cycle. On the infrastructure side, google-cloud-python fixed an mTLS test failure by properly mocking os.path.exists() to align with stricter google-auth discovery checks [5], and shipped a thread-safe Marshal initialization to prevent dictionary-changed-size-during-iteration crashes in Firestore on_snapshot scenarios [6]. JAX also landed Mosaic GPU improvements: bounds checking for profiler event writes to prevent kernel crashes from SMEM corruption , and a OneOf constraint solver for layout inference that handles user-explicit index casting .

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] Get rid of the plethora of async collective primitives we have. We can add more as needed. ↗ google/jax
  2. [2] Parameterize sampler tests by whether or not to jit ↗ google/jax
  3. [3] v2.13.0 ↗ googleapis/python-genai
  4. [4] fix: add deprecation warnings to Imagen generate_images, edit_images, generate_videos (if using prompt/text/image args), LiveConnectConfig.GenerationConfig which will be removed in the next major vers googleapis/python-genai
  5. [5] fix(gapic): mock os.path.exists in mTLS tests to support newer google auth (#17807) googleapis/google-cloud-python
  6. [6] fix(proto-plus): make Marshal thread-safe and handle race conditions ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on July 22, 2026?
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. In total, 48 commits, 47 pull requests, and 5 releases landed.
Who contributed to Google on July 22, 2026?
7 developers shipped this update, including copybara-service, samanklesaria, Patrick C. Toulme, release-please[bot], Amy Wu, Chalmer Lowe, and Anthonios Partheniou.
What were the notable Google updates?
Get rid of the plethora of async collective primitives we have. We can add more as needed, Parameterize sampler tests by whether or not to jit, and v2.13.0.

More from Google

Daily updates, in your inbox

Follow Google

What actually shipped in Google, written up every day — commits, pull requests, releases, and security advisories.

or

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

Elsewhere on the wire

Want every project, not just this one?