RepoJournal
Google

Google

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

Pick a date

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

The Wire · Showcase

JAX BAKES ROCM INTO WHEELS, GENAI BREAKS AFC, CLOUD PYTHON SHIPS

By RepoJournal · Filed · About Google

JAX wheels now embed ROCm runtime paths at link time, killing the post-build patchelf dance and fixing CI against TheRock's non-standard layout.

The JAX team consolidated ROCm wheel RUNPATH strategy by embedding it at link time via _WHEEL_RPATHS instead of patching after the build [1]. Single wheels now resolve ROCm libs across TheRock pip wheels, tarballs under /opt/rocm-<ver>, and legacy /opt/rocm without the loader stumbling over $ORIGIN entries. Meanwhile, Pallas/Mosaic GPU docs got a hyperlink overhaul that also corrects stale API names like commit_group to commit_smem_to_gmem_group and removes the defunct plgpu.TransposeTransform [2]. Pallas MGPU now supports N-dimensional TMEM references instead of just 2D by introducing CollapseLeadingBatchDimensionsTransform to collapse batch dimensions into the minor axis [3]. Over in python-genai, the team announced an upcoming breaking change to Automatic Function Calling (AFC) and swapped deprecated .copy() calls for .model_copy() [4], [5]. google-cloud-python is shipping a new release via Librarian with gapic-generator v1.35.0 [6], and Firestore system tests got hammered down from one hour to 13 minutes by adding pytest-xdist parallelism, module-scoped fixtures, and faster polling [7].

Action items

References

  1. [1] [ROCm] Bake TheRock-aware RUNPATH into wheels and fix /opt/rocm-less CI ↗ google/jax
  2. [2] [Pallas/Mosaic GPU][Docs] Add hyperlinks to various API bits throughout the reference. google/jax
  3. [3] [Pallas:MGPU] Support batch dimension for TMEM references ↗ google/jax
  4. [4] docs: announce Automatic Function Calling (AFC) upcoming breaking change warning googleapis/python-genai
  5. [5] fix: use .model_copy() instead of deprecated .copy() googleapis/python-genai
  6. [6] chore: librarian release pull request: 20260611T192009Z (#17432) googleapis/google-cloud-python
  7. [7] chore(firestore): optimize system tests runtime ↗ googleapis/google-cloud-python

FAQ

What changed in Google on June 12, 2026?
JAX wheels now embed ROCm runtime paths at link time, killing the post-build patchelf dance and fixing CI against TheRock's non-standard layout.
What should Google teams do about it?
Review AFC breaking change warning in python-genai before upgrading • Update JAX wheels if you're running against non-standard ROCm layouts • Monitor google-cloud-python release rollout for compatibility
Which Google repositories shipped on June 12, 2026?
google/jax, googleapis/python-genai, googleapis/google-cloud-python

Related across the cluster

For your repos

The showcase is a teaser.
Your wire is the product.

Same engine. Different stack. Below: what changes when the wire is yours.

Showcase wire

  • •14 famous open source orgs
  • •One wire per day
  • •Public, generic
  • •Read on the web, when you remember

Your wire

  • →Up to 1,500 of your repos - orgs, deps, vendors
  • →Morning and evening briefs
  • →Action items routed to your team
  • →Slack delivery, email, breaking-news CVE alerts

Want a hands-on demo first? Ask a current user for an invite link.