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 GPU KERNELS GET FASTER MEMORY ACCESS, BIGFRAMES UNLOCKS PANDAS-STYLE COLUMN INDEXING

By RepoJournal · Filed · About Google

JAX's Mosaic GPU backend shipped support for specialized matrix load/store instructions that will cut memory bottlenecks on supported hardware, while BigFrames now lets you access DataFrame columns by position like you're working with pandas.

The Mosaic GPU team landed ldmatrix/stmatrix instruction support [1], clearing the path for sharded address calculations that'll meaningfully improve memory throughput on tensor operations. This is the foundation work; the full performance win comes in the follow-up when they enable num>1 sharding, but shipping this now keeps the pipeline moving without blocking downstream work. On the JAX core side, register_hlo_module_transformation got refactored to support AOT compilation [2] by pulling PJRT_Api directly from platforms instead of routing through clients that don't exist in ahead-of-time contexts. This unblocks compiling JAX graphs without initializing a runtime, a critical path for edge deployment and mobile. BigFrames shipped offset-based column access via iloc [3], which means your team can finally use familiar pandas patterns like df.iloc[:, 0] instead of remembering column names. Small QOL win but it closes a friction point for teams migrating from pandas workflows.

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] [Mosaic GPU] Add basic support for ldmatrix/stmatrix ↗ google/jax
  2. [2] Refactor register_hlo_module_transformation to support AOT compilation ↗ google/jax
  3. [3] feat(bigframes): support offset-based column access via iloc ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on July 11, 2026?
JAX's Mosaic GPU backend shipped support for specialized matrix load/store instructions that will cut memory bottlenecks on supported hardware, while BigFrames now lets you access DataFrame columns by position like you're working with pandas. In total, 40 commits and 33 pull requests landed.
Who contributed to Google on July 11, 2026?
7 developers shipped this update, including copybara-service, Yash Katariya, skye-ant, Allan Renucci, mattjj, Wanlin Du, and tswast.
What were the notable Google updates?
[Mosaic GPU] Add basic support for ldmatrix/stmatrix, Refactor register_hlo_module_transformation to support AOT compilation, and feat(bigframes): support offset-based column access via iloc.

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?

We use privacy-friendly analytics (Google Analytics, IP-anonymized) to see which pages help readers. No ads, and we never sell your data. See our Privacy Policy.