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
- → Pull JAX main to pick up the Mosaic 64-bit fix if you're using Pallas for on-device compilation google/jax [plan]
- → Monitor google-cloud-python for the Librarian refresh to land in your next release cycle googleapis/google-cloud-python [monitor]
References
- [1] [pallas] Fix Pallas Mosaic 64-bit mode loop indexing ↗ google/jax
- [2] Update XLA dependency to use revision http://github.com/openxla/xla/commit/93f41f404fa2e38df46cd8cf8220613ca5182c48 google/jax
- [3] chore: update librarian to v0.14.1-0.20260522151051-433d9e6211ad ↗ googleapis/google-cloud-python
- [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