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 DROPS PYTHON 3.13T SUPPORT, CLEANS UP BUILD PIPELINE

By RepoJournal · Filed · About Google

JAX consolidated its dependency lock files after dropping Python 3.13 free-threading support, signaling a shift in the framework's build strategy as it pushes through infrastructure modernization.

The consolidation removes the need for separate lock files between freethreading and non-freethreading builds [1], simplifying the build process after Python 3.13t reached end-of-life for JAX's purposes. In parallel, the team bumped XLA and rules_python to v2.2.0 [2] [3], removing temporary patches that were holding back the build system. JAX also made dce_sink a VJPHiPrimitive to unlock distributed training capabilities [4], a critical upgrade for sharding-dependent workloads. Over at google-cloud-python, the pace is accelerating: Bigtable samples got reorganized by client type [5], client-side metrics handlers shipped [6], and compatibility tests were added to catch regressions [7]. Meanwhile, python-genai is optimizing for speed, deferring model validator construction from import time to first use [8] [9], which cuts startup overhead for applications that don't immediately need validation. The auth layer also got hardened with fallback support for X.509 workload identity on ECP machines [10], preventing TypeErrors in certificate path resolution.

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] Consolidate freethreading and non-freethreading requirements lock files. google/jax
  2. [2] Bump the XLA commit and remove the temporary rules_python_220.patch. google/jax
  3. [3] Bump rules_python to v2.2.0. ↗ google/jax
  4. [4] Make dce_sink a VJPHiPrimitive so that it works with sharding ↗ google/jax
  5. [5] chore(bigtable): reorganize samples ↗ googleapis/google-cloud-python
  6. [6] feat(bigtable): client side metrics handlers ↗ googleapis/google-cloud-python
  7. [7] tests(bigtable): System/compatibility tests for testing shim compatiblity ↗ googleapis/google-cloud-python
  8. [8] perf: build model validators on first use instead of at import ↗ googleapis/python-genai
  9. [9] perf: build model validators on first use instead of at import googleapis/python-genai
  10. [10] fix(auth): prevent TypeError and support home-dir cert fallback for X… ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on August 12, 2026?
JAX consolidated its dependency lock files after dropping Python 3.13 free-threading support, signaling a shift in the framework's build strategy as it pushes through infrastructure modernization. In total, 49 commits and 42 pull requests landed.
Who contributed to Google on August 12, 2026?
8 developers shipped this update, including Peter Hawkins, copybara-service, Benjamin Chetioui, Amy Wu, Google Team Member, daniel-sanche, hebaalazzeh, and attharva-24.
What were the notable Google updates?
Consolidate freethreading and non-freethreading requirements lock files, Bump the XLA commit and remove the temporary rules_python_220.patch, and Bump rules_python to v2.2.0.

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?