RepoJournal
Google

Google

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

Keep up with Google in about 3 minutes a day: what actually shipped — the commits, pull requests, releases, and security advisories that matter.

One email a day. Unsubscribe in one click.

Pick a date

Topics: AI / ML Python Full archive →

The Wire · Showcase

JAX EMERGENCY ROLLBACKS ACROSS THREE COMMITS

By RepoJournal · Filed · About Google

Three consecutive commits hit the revert button overnight, signaling a critical issue that slipped through in the experimental Pallas subsystem.

JAX shipped three separate rollbacks in rapid succession [1][2][3], suggesting a cascading problem in recent changes that made it to main. The team is also deprecating the old jax.experimental.pallas debug API in favor of the config-based alternative [4], which means code using the legacy enable_debug_checks and debug_checks_enabled paths will need updates before the next major release. Cleanup continued with the removal of obsolete libtpu version checks in TPU Pallas tests , indicating the team is tightening the test suite as they stabilize this infrastructure layer. No public issue statements yet, but three rollbacks in 24 hours means something broke badly enough that keeping it landed was worse than losing the feature.

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] Reverts 6a4f44085525c2f72ef291f19453b2e000a299a8 ↗ google/jax
  2. [2] rollback ↗ google/jax
  3. [3] rollback ↗ google/jax
  4. [4] Deprecate jax.experimental.pallas.{enable_debug_checks,debug_checks_enabled} google/jax

Quick answers

What shipped in Google on May 17, 2026?
Three consecutive commits hit the revert button overnight, signaling a critical issue that slipped through in the experimental Pallas subsystem. In total, 6 commits and 5 pull requests landed.
Who contributed to Google on May 17, 2026?
1 developer shipped this update, including copybara-service[bot].
What were the notable Google updates?
Reverts 6a4f44085525c2f72ef291f19453b2e000a299a8, rollback, and rollback.

More from Google

Daily updates, in your inbox

Follow Google

Keep up with Google in about 3 minutes a day: what actually shipped — the commits, pull requests, releases, and security advisories that matter.

or

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

Elsewhere on the wire

Want every project, not just this one?