Who contributed to Google on May 31, 2026?

2 developers shipped this update, including mattjj and copybara-service[bot].

What were the notable Google updates?

restore triangular_solve batching rule, improve sharding rule and Support while_p in GPU Interpret Mode.

♥

May 31, 2026

$ the-wire · showcase

JAX FIXES TRIANGULAR_SOLVE MEMORY BLOWUP, EXPANDS GPU INTERPRET MODE

By RepoJournal · Filed 06:02 UTC on May 31, 2026 · About Google · Composed from the cited sources · methodology

2 people shipped this

mattjj @mattjj 1 cited

copybara-service[bot] @copybara-service[bot] 1 cited

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

→ Sync JAX to latest if you hit triangular_solve OOMs google/jax [immediate]
→ Test GPU Interpret Mode with while_p workloads if debugging GPU kernels google/jax [plan]

References

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

$ ls google/reviews/ # the step-back reads

the week in review JAX drops legacy IFRT device APIs, google-auth adds 3.15 the month in review JAX free-threading push reshapes 0.11.2 and 0.12 cleanup

Full archive → Weekly reviews →

Quick answers

What shipped 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. In total, 4 commits and 2 pull requests landed.
Who contributed to Google on May 31, 2026?: 2 developers shipped this update, including mattjj and copybara-service[bot].
What were the notable Google updates?: restore triangular_solve batching rule, improve sharding rule and Support while_p in GPU Interpret Mode.

♥

$ log Google --recent

2026-10-04 JAX pins OpenXLA revision, disables overlap_test against older libtpu 5 shipped 2026-10-03 JAX adds type-2 and type-3 Hermitian eigensolves, lexsort batching cuts unique compile time 69 shipped 2026-10-02 JAX hijax flags read at call time, Spanner stops deleting multiplexed sessions 112 shipped 2026-10-01 Vertex AI client timeout fix, Compute v1 API regenerated 87 shipped

$ log --all # elsewhere on the wire

@vuejs 2026-10-04 Vue language tools stop calling registerHooks on buggy Node builds @openclaw 2026-10-04 OpenClaw's deslop sweep removes 1,000+ production lines

$ ls showcases/

$ init your-desk

RepoJournal can write one of these for any repo you care about. Sign in with GitHub, pick your repos and sources, and your desk gets a wire like this one every morning.