RepoJournal
Google

Google

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

Pick a date

  1. Sat 30 may
  2. Sun 31 may
  3. Mon 1 jun
  4. Mon 8 jun
  5. Tue 9 jun
  6. Wed 10 jun
  7. Thu 11 jun
  8. Fri 12 jun
  9. Sat 13 jun
  10. Sun 14 jun
  11. Mon 15 jun
  12. Tue 16 jun
  13. Wed 17 jun
  14. Thu 18 jun
  15. Fri 19 jun
  16. Sat 27 jun
  17. Sun 28 jun
  18. Mon 29 jun
  19. Tue 30 jun
  20. Wed 1 jul
  21. Thu 2 jul
  22. Fri 3 jul
  23. Sat 4 jul
  24. Sun 5 jul
  25. Mon 6 jul
  26. Tue 7 jul
  27. Wed 8 jul
  28. Thu 9 jul
  29. Fri 10 jul
  30. Today 11 jul

The Wire · Showcase

JAX GPU KERNELS GET FASTER MEMORY ACCESS, BIGFRAMES UNLOCKS PANDAS-STYLE COLUMN INDEXING

By RepoJournal · Filed · About Google

copybara-service Yash Katariya skye-ant Allan Renucci mattjj Wanlin Du tswast
7 people shipped this

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.

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

More from Google

Google Google

JAX FINALIZES CORE API DEPRECATIONS WHILE FP8 MIXED PRECISION LANDS ON MOSAIC GPU

JAX is cleaning house on jax.core APIs while shipping native mixed E4M3/E5M2 FP8 support that enterprises using Mosaic GPUs have been waiting for.

ai 66 shipped 1-min read
Google Google

JAX FIXES CUSTOM_VJP3 AUTODIFF BUGS; PYTHON-GENAI BREAKS BACKWARD COMPATIBILITY

JAX shipped critical fixes to custom_vjp3 that were producing wrong-shaped outputs under vmap-of-jvp, while python-genai landed a breaking change removing three parameters and reshaping safety settings.

ai 72 shipped 1-min read
Google Google

JAX ADDS ONEAPI GPU RUNTIME AS PYTHON 3.14 FORCES GOOGLE-AUTH DEPENDENCY PURGE

JAX shipped OneAPI (SYCL) GPU support while google-auth spent the night locking down five critical dependencies for Python 3.14 compatibility.

ai 55 shipped 1-min read
Google Google

JAX REFACTORS CORE COMPILER PIPELINE FOR HARDWARE EFFICIENCY

JAX is reshaping its Pallas and Mosaic layers to delay grid partitioning until lowering time, unlocking parallelism that batching operations couldn't reach before.

ai 89 shipped 1-min read

Daily updates, in your inbox

Follow Google

JAX, the GenAI SDK, and the Cloud libs — Google's open source layer We'll email you a link to confirm first.

Free. Confirm via email. Unsubscribe in one click.

— or follow the whole beat:

Elsewhere on the wire

Elixir & Phoenix Elixir & Phoenix

ELIXIR FIXES TIME MATH BUG THAT ROUNDS THE WRONG WAY

NaiveDateTime.diff/3 has been silently rounding elapsed time up instead of down, breaking its documented contract and potentially causing logic errors in any code relying on time-based calculations.

Alexander Gubarev nseaSeb Łukasz Samson SteffenDE
elixir 35 shipped 1-min read
.NET @dotnet

OPERATOR SDK 13.0 BREAKS CONTROLLER ARCHITECTURE, ANDROID TESTS GO LOCAL

The dotnet-operator-sdk ships a major breaking change that fundamentally rewrites how multiple controllers target the same entity, while the Android team closes external test dependencies to fix certificate export failures on macOS.

github-actions[bot] kimpenhaus renovate Weihan Li +7
dotnet 40 shipped 2-min read
Spring @spring-projects

SPRING INTEGRATION FIXES CRITICAL LOCK REGISTRY BUG BLOCKING DISTRIBUTED SYSTEMS

Spring Integration shipped an emergency fix for JdbcLockRegistry that was causing false application hangs when different keys collided on shared locks.

dependabot artembilan Oliver Drotbohm
java 10 shipped 1-min read
FastAPI & Pydantic FastAPI & Pydantic

PYDANTIC ADDS VALIDATION ERROR TROUBLESHOOTING GUIDE

Pydantic just shipped a troubleshooting guide that walks you through reading validation failures in Logfire traces, cutting your debugging time in half.

strawgate
python 2 shipped 1-min read