The Wire · Showcase
JAX OVERHAULS GPU BUILD SYSTEM; GENAI SHIPS BLOCKING FUNCTION CALLS
By RepoJournal · Filed · About Google
JAX fixed critical Bzlmod build failures across Windows, CUDA, and ROCm while python-genai v2.1.0 lands server-side tool deltas and video response parameters.
The JAX team shipped a comprehensive fix [2] for Bzlmod builds that restores Windows and GPU compilation after curl patches became unnecessary and cross-compile toolchains stopped working with the new build system. Separately, they relaxed Mosaic GPU's overly strict divisibility constraints on gather index operations [1], unblocking production workloads that had been constrained by unnecessary alignment requirements. The team also renamed `auto_pvary` to `auto_pcast` [3] to better reflect its actual behavior and exposed the config flag, tidying up the public API. A critical Mosaic NCCL rollback [4] landed after breaking OSS tests, so if you're tracking main, revert and wait for the fixed version. Over in python-genai, v2.1.0 dropped with blocking FunctionCall support in Live API [5], server-side tool deltas [6], and gemini-3.1-flash-lite model support [7]. The google-cloud-python team simultaneously dropped Python 3.9 support [8] across api-core and removed it from default nox sessions [10], forcing upgrades for teams still on older runtimes. SQLAlchemy 2.0 compatibility fixes [11] and pandas test cleanup [9] round out the week's stability work.
Action items
- → Verify JAX Bzlmod Windows/CUDA/ROCm builds work in your CI — if you use GPU, test immediately google/jax [immediate]
- → Update python-genai to v2.1.0 if you use Live API with function calling googleapis/python-genai [plan]
- → Upgrade to Python 3.10+ for google-api-core; Python 3.9 is no longer tested googleapis/google-cloud-python [plan]
- → Monitor JAX main for Mosaic NCCL re-fix; current version has known regression google/jax [monitor]
References
- [1] [Mosaic GPU] Relax the divisibility constraint for loads/stores with gather indices. ↗ google/jax
- [2] Fixes for JAX Bzlmod builds ↗ google/jax
- [3] * Rename `auto_pvary` to `auto_pcast` since it disables more than just pvarys. ↗ google/jax
- [4] [Mosaic:GPU] Rollback: Use NCCL API on the host to collective peer addresses. ↗ google/jax
- [5] feat: support Blocking FunctionCall in Live API in AgentPlatform (Vertex) ↗ googleapis/python-genai
- [6] feat: Introduce Server Side tools deltas ↗ googleapis/python-genai
- [7] v2.1.0 ↗ googleapis/python-genai
- [8] fix(api-core): drop support for Python 3.9 (#16980) googleapis/google-cloud-python
- [9] test: remove pandas version restrictions and fix precision issue (#17005) googleapis/google-cloud-python
- [10] chore: remove Python 3.9 from default nox session (#17017) googleapis/google-cloud-python
- [11] fix(sqlalchemy-bigquery): update literal binds test for SQLAlchemy 2.0 (#17029) googleapis/google-cloud-python
FAQ
- What changed in Google on May 12, 2026?
- JAX fixed critical Bzlmod build failures across Windows, CUDA, and ROCm while python-genai v2.1.0 lands server-side tool deltas and video response parameters.
- What should Google teams do about it?
- Verify JAX Bzlmod Windows/CUDA/ROCm builds work in your CI — if you use GPU, test immediately • Update python-genai to v2.1.0 if you use Live API with function calling • Upgrade to Python 3.10+ for google-api-core; Python 3.9 is no longer tested
- Which Google repositories shipped on May 12, 2026?
- google/jax, googleapis/python-genai, googleapis/google-cloud-python