RepoJournal
Google

Google

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

Pick a date

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

The Wire · Showcase

JAX FIXES TRIANGULAR_SOLVE MEMORY BLOWUP, EXPANDS GPU INTERPRET MODE

By RepoJournal · Filed · About Google

A partial revert of yesterday's batching rule is rolling back OOMs in triangular_solve while keeping the performance gains.

JAX shipped a fix [1] that reverts the problematic broadcast insertion from the previous batching rule but preserves the optimization by improving the sharding rule instead. That earlier PR was causing out-of-memory errors in certain batching cases, so this lands the best of both worlds: the batching rule stays lean while the sharding rule now accepts additional batch dimensions. In parallel, GPU Interpret Mode got expanded [2] to support while_p operations and better handle unpacking of arbitrary transforms for copy_gmem_to_smem_p, which broadens what workloads can now run in this debugging mode. These are targeted fixes addressing real memory and compatibility pain points from the last cycle.

Action items

References

  1. [1] restore triangular_solve batching rule, improve sharding rule ↗ google/jax
  2. [2] Support while_p in GPU Interpret Mode. ↗ google/jax

FAQ

What changed in Google on May 31, 2026?
A partial revert of yesterday's batching rule is rolling back OOMs in triangular_solve while keeping the performance gains.
What should Google teams do about it?
Sync JAX to latest if you hit triangular_solve OOMs • Test GPU Interpret Mode with while_p workloads if debugging GPU kernels
Which Google repositories shipped on May 31, 2026?
google/jax

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.