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.
One email a day. Unsubscribe in one click.
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. Read a past issue →
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