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 Renames Mosaic GPU Parameters

By RepoJournal · Filed · About Google

Google's JAX team is renaming a core Mosaic GPU attribute, and it will break your cross-process parameter code.

Google's JAX team is renaming the `multimem_parameters` Mosaic attribute to `symmetric memory parameters`, a change that will impact anyone using Mosaic GPU for cross-process or multimem workloads [1]. The attribute marks parameters that "should be copied to the symmetrical memory space," so the rename clarifies intent but will require updating your code [2]. The change landed via two commits, one by copybara-service and one by a human author, with the latter including the PiperOrigin-RevId [2]. Meanwhile, the XLA dependency was bumped to a specific revision, likely to keep pace with the GPU backend changes [3]. If you're on Mosaic GPU, plan to update your parameter annotations before your next deploy.

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] [Mosaic:GPU] Rename multimem parameters mosaic attribute to symmetric memory parameters. ↗ google/jax
  2. [2] [Mosaic:GPU] Rename multimem parameters mosaic attribute to symmetric memory parameters. google/jax
  3. [3] Update XLA dependency to use revision http://github.com/openxla/xla/commit/ddde0c9ee5a4aa767ff35af8e05473117c7028e4 google/jax

Quick answers

What shipped in Google on August 23, 2026?
Google's JAX team is renaming a core Mosaic GPU attribute, and it will break your cross-process parameter code. In total, 2 commits and 1 pull requests landed.
Who contributed to Google on August 23, 2026?
3 developers shipped this update, including copybara-service, Levon Ter-Grigoryan, and jax authors.
What were the notable Google updates?
[Mosaic:GPU] Rename multimem parameters mosaic attribute to symmetric memory parameters, [Mosaic:GPU] Rename multimem parameters mosaic attribute to symmetric memory parameters, and Update XLA dependency to use revision http://github.com/openxla/xla/commit/ddde0c9ee5a4aa767ff35af8e05473117c7028e4.

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?