RepoJournal
Google

Google

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

Pick a date

The Wire · Showcase

JAX FIXES 64-BIT MOSAIC LOOP INDEXING, GOOGLE CLOUD PYTHON REFRESHES LIBRARIAN

By RepoJournal · Filed · About Google

Pallas Mosaic now runs reliably in 64-bit mode after a critical type mismatch fix that was blocking pipeline operations.

JAX shipped a fix for Pallas Mosaic's 64-bit mode that resolves a type promotion bug in pipeline loops [1]. The error occurred when operations got promoted to 64-bit inside the loop, causing arith.cmpi operations to fail due to mismatched operand types. This lands alongside an XLA dependency bump [2] that keeps JAX's compiler foundation current. On the Cloud Python side, google-cloud-python updated Librarian to v0.14.1 [3], the code generation tool that produces the SDK's client libraries. The same team simultaneously refactored template generation to eliminate duplication [4], reducing the surface area for maintenance debt. These are housekeeping moves, but they clear the way for cleaner API surface generation and fewer template variants to keep in sync.

Action items

References

  1. [1] [pallas] Fix Pallas Mosaic 64-bit mode loop indexing ↗ google/jax
  2. [2] Update XLA dependency to use revision http://github.com/openxla/xla/commit/93f41f404fa2e38df46cd8cf8220613ca5182c48 google/jax
  3. [3] chore: update librarian to v0.14.1-0.20260522151051-433d9e6211ad ↗ googleapis/google-cloud-python
  4. [4] chore: refactor templates to reduce duplication ↗ googleapis/google-cloud-python

FAQ

What changed in Google on May 26, 2026?
Pallas Mosaic now runs reliably in 64-bit mode after a critical type mismatch fix that was blocking pipeline operations.
What should Google teams do about it?
Pull JAX main to pick up the Mosaic 64-bit fix if you're using Pallas for on-device compilation • Monitor google-cloud-python for the Librarian refresh to land in your next release cycle
Which Google repositories shipped on May 26, 2026?
google/jax, googleapis/google-cloud-python

Related across the cluster

For your repos

The showcase is a teaser.
Your wire is the product.

Same engine. Different stack. Below: what changes when the wire is yours.

Showcase wire

  • 14 famous open source orgs
  • One wire per day
  • Public, generic
  • Read on the web, when you remember

Your wire

  • Up to 1,500 of your repos - orgs, deps, vendors
  • Morning and evening briefs
  • Action items routed to your team
  • Slack delivery, email, breaking-news CVE alerts

Want a hands-on demo first? Ask a current user for an invite link.