RepoJournal
Google

Google

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

What actually shipped in Google, written up every day — commits, pull requests, releases, and security advisories.

One email a day. Unsubscribe in one click.

Pick a date

  1. Sat 13 jun
  2. Sun 14 jun
  3. Mon 15 jun
  4. Tue 16 jun
  5. Wed 17 jun
  6. Thu 18 jun
  7. Fri 19 jun
  8. Sat 27 jun
  9. Sun 28 jun
  10. Mon 29 jun
  11. Tue 30 jun
  12. Wed 1 jul
  13. Thu 2 jul
  14. Fri 3 jul
  15. Sat 4 jul
  16. Sun 5 jul
  17. Mon 6 jul
  18. Tue 7 jul
  19. Wed 8 jul
  20. Thu 9 jul
  21. Fri 10 jul
  22. Sat 11 jul
  23. Sun 12 jul
  24. Mon 13 jul
  25. Tue 14 jul
  26. Wed 15 jul
  27. Thu 16 jul
  28. Sat 18 jul
  29. Sun 19 jul
  30. Mon 20 jul
Topics: AI / ML Python Full archive →

The Wire · Showcase

JAX TIGHTENS PPERMUTE VALIDATION TO CATCH SILENT CORRECTNESS BUGS

By RepoJournal · Filed · About Google

mattjj copybara-service kodlan Yash Katariya jax authors
5 people shipped this
mattjj @mattjj 1 cited copybara-service @copybara-service 1 cited kodlan @kodlan 1 cited Yash Katariya @yashk2810 jax authors @Google-ML-Automation

JAX is shipping a correctness fix that will break code relying on undefined behavior, and that's exactly the point.

The ppermute collective operation has been silently producing wrong answers when axis names don't match mesh axis order [1]. A new flag, jax_raise_on_ppermute_sort_diff, is enabled by default and will error instead of auto-sorting replica groups. This catches bugs that have likely shipped to production undetected. If your distributed training uses ppermute, expect this to surface issues in your axis naming. Separately, JAX documented the sum_match argument for jax.vmap [2], clarifying that outputs varying along the mapped axis get summed in autodiff contexts instead of erroring. Documentation improvements landed for structured residuals and first-class VJPs [3], making custom derivative patterns less opaque for teams building on JAX's autodiff.

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] Add `jax_raise_on_ppermute_sort_diff` enabled by default which errors out if axis_names passed to ppermute don't match the mesh axis_names order. ↗ google/jax
  2. [2] Document the sum_match argument of jax.vmap ↗ google/jax
  3. [3] [docs] add structured-residuals docs to the 301 pages ↗ google/jax

Quick answers

What shipped in Google on July 20, 2026?
JAX is shipping a correctness fix that will break code relying on undefined behavior, and that's exactly the point. In total, 5 commits and 4 pull requests landed.
Who contributed to Google on July 20, 2026?
5 developers shipped this update, including mattjj, copybara-service, Yash Katariya, jax authors, and kodlan.
What were the notable Google updates?
Add `jax_raise_on_ppermute_sort_diff` enabled by default which errors out if axis_names passed to ppermute don't match the mesh axis_names order, Document the sum_match argument of jax.vmap, and [docs] add structured-residuals docs to the 301 pages.

More from Google

Google Google

JAX BACKWARD PASS NOW LOGS DEBUG DATA. CLOUD AUTH TIGHTENS mTLS PRECEDENCE

JAX's autodiff pipeline shipped structured logging from transpose rules, letting you inspect what's happening inside the backward pass without tanking performance.

mattjj nbayati
ai 15 shipped 1-min read
Google Google

JAX ACCELERATES MULTI-GPU COMPUTE WHILE GOOGLE AUTH PLUGS CONNECTION LEAK

JAX shipped three critical compiler improvements for distributed training while google-auth patches a file descriptor exhaustion bug that could starve production services.

copybara-service Yash Katariya Gattocrucco Google Team Member +3
ai 68 shipped 2-min read
Google Google

JAX ADDS FP6 SUPPORT END-TO-END, PYTHON-GENAI SHIPS AGENT CONFIG

JAX landed full float6 format support across CPU and GPU backends, while the AI SDK gains AntigravityAgentConfig and a trigger resource in v2.12.0.

copybara-service mattjj release-please[bot] release-please +4
ai 88 shipped 1-min read
Google Google

JAX FIXES ROCM TRITON SUPPORT, CLOUD PYTHON TIGHTENS IMPORT PERFORMANCE

JAX's Triton integration on AMD GPUs is finally working after resolving a device architecture detection bug that broke every pallas_call on ROCm hardware.

copybara-service magaonka-amd Wanlin Du noahdietz +1
ai 75 shipped 1-min read

Daily updates, in your inbox

Follow Google

What actually shipped in Google, written up every day — commits, pull requests, releases, and security advisories.

or

One email a day. Unsubscribe in one click. Read a past issue →

Elsewhere on the wire

CachyOS @cachyos

LINUX 7.2-RC4 LANDS WITH GAMING SCHEDULER, ANANICY OVERHAULS PROCESS TUNING

CachyOS shipped kernel 7.2-rc4 with a new gaming-optimized scheduler branch and AMD display fixes, while ananicy-rules yanked normal scheduling from most process types in a major behavioral shift.

Eric Naim Aarrayy Luka Ogadze Daniel Lindestad +2
infra 56 shipped 1-min read
Elixir & Phoenix Elixir & Phoenix

ELIXIR COMPILER GETS 20% FASTER TYPE CHECKING

Multi-clause function analysis now runs significantly faster after the core team eliminated unnecessary list allocations in disjoint argument checking.

Alexander Gubarev preciz José Valim
elixir 6 shipped 1-min read
.NET @dotnet

MACIOS FIXES SYMBOL UPLOAD GAP, MSBUILD PULLS ARCADE UPDATES

Apple's App Store Connect will finally see proper crash symbolication for .NET iOS apps after macios shipped a fix that embeds symbol files directly in the .ipa.

dotnet-maestro dependabot Jan Provazník rolfbjarne +2
dotnet 21 shipped 1-min read
FastAPI & Pydantic FastAPI & Pydantic

PYDANTIC DOCS GET GRAMMAR CLEANUP

Pydantic merged a four-typo fix across docs and skill definitions that slipped past CI.

chinesepowered
python 2 shipped 1-min read
Want every project, not just this one?