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.
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. Read a past issue →
Action items
- → Review JAX build changes and update your local development environment to use consolidated lock files google/jax [plan]
- → Verify Bigtable sample references if you depend on the old directory structure googleapis/google-cloud-python [monitor]
- → Test python-genai import performance in your application after validator optimization googleapis/python-genai [monitor]
References
- [1] Consolidate freethreading and non-freethreading requirements lock files. google/jax
- [2] Bump the XLA commit and remove the temporary rules_python_220.patch. google/jax
- [3] Bump rules_python to v2.2.0. ↗ google/jax
- [4] Make dce_sink a VJPHiPrimitive so that it works with sharding ↗ google/jax
- [5] chore(bigtable): reorganize samples ↗ googleapis/google-cloud-python
- [6] feat(bigtable): client side metrics handlers ↗ googleapis/google-cloud-python
- [7] tests(bigtable): System/compatibility tests for testing shim compatiblity ↗ googleapis/google-cloud-python
- [8] perf: build model validators on first use instead of at import ↗ googleapis/python-genai
- [9] perf: build model validators on first use instead of at import googleapis/python-genai
- [10] fix(auth): prevent TypeError and support home-dir cert fallback for X… ↗ googleapis/google-cloud-python