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 TIGHTENS MOSAIC BOUNDS CHECKING AS GENAI GAINS MULTIMODAL FILE SEARCH

By RepoJournal · Filed · About Google

JAX shipped a critical bounds verification fix in Mosaic matmul operations while the Python GenAI SDK adds embeddings-powered file search with media download support.

The JAX team merged bounds checks for dimension indices in `MatmulOp::verify` [1], closing a gap where invalid indices from `DotDimensionNumbersAttr` could trigger array access violations on TPU operations. This lands alongside an XLA dependency bump [2] that pulls in upstream optimization work. On the GenAI front, Python SDK now ships multimodal file search capabilities [3], adding embedding model selection for file store creation, media ID tracking in grounding chunks, and a new `file_search_stores.downloadMedia()` method for retrieving retrieved content. The file search enhancement targets RAG workflows where vector embeddings need to index unstructured media alongside text. JAX users running Mosaic-heavy workloads should merge the bounds fix; GenAI SDK consumers using file stores should pull the update to unlock download functionality.

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] [Mosaic] Add bounds checks for dimension indices ↗ google/jax
  2. [2] Update XLA dependency to use revision http://github.com/openxla/xla/commit/341f69b4e5505a52e444780c52d05b8790a39df1 google/jax
  3. [3] feat: [Python] Multimodal file search ↗ googleapis/python-genai

Quick answers

What shipped in Google on May 4, 2026?
JAX shipped a critical bounds verification fix in Mosaic matmul operations while the Python GenAI SDK adds embeddings-powered file search with media download support. In total, 3 commits and 2 pull requests landed.
Who contributed to Google on May 4, 2026?
1 developer shipped this update, including copybara-service[bot].
What were the notable Google updates?
[Mosaic] Add bounds checks for dimension indices, Update XLA dependency to use revision http://github.com/openxla/xla/commit/341f69b4e5505a52e444780c52d05b8790a39df1, and feat: [Python] Multimodal file search.

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?