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 CUTS THREAD OVERHEAD IN GPU KERNELS WHILE PYTHON-GENAI SHIPS RC1 WITH NEW MODEL SUPPORT

By RepoJournal · Filed · About Google

JAX's Pallas GPU kernel interpreter is now thread-efficient, cutting unnecessary jit compilations that were exhausting thread pools.

The JAX team shipped a significant optimization to Pallas [1] that removes redundant conversions to JAX arrays in host callbacks, directly addressing thread pool exhaustion during jitting. This pairs with version cleanup [2] that bumps jaxlib to 0.10.2, streamlining the codebase after the recent stable release. On the GPU side, cuDNN SDPA flex attention is back online with a version guard for 9.13+ [3] to prevent NaN issues that plagued earlier builds. Meanwhile, python-genai dropped v2.9.0-rc1 [4] with audioOffset support for VoiceActivity and local tokenizer mappings for gemini-3-flash-preview, gemini-3.1, gemini-3.5, and gemini-4 [5]. In the cloud Python ecosystem, google-cloud-python reverted a mypy centralization experiment [6] after early feedback, while pubsub shipped a cleanup pass removing Python 3.7/3.8/3.9 compatibility code [7] now that the library requires 3.10+. CI infrastructure got a refresh [8] to address Node.js 20 deprecation warnings ahead of the June 16th deadline.

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] [Pallas][GPU kernel interpreter] Remove explicit conversions to Jax arrays in host callbacks. google/jax
  2. [2] Remove stale version guards after JAX v0.10.2 release. google/jax
  3. [3] Reenable cuDNN SDPA flex attention with version guard >= 9.13 to avoid NaNs ↗ google/jax
  4. [4] v2.9.0-rc1 ↗ googleapis/python-genai
  5. [5] feat: Add gemini-3-flash-preview (gemini-3.1, gemini-3.5, gemini-4 are already mapped) to the local tokenizer map. googleapis/python-genai
  6. [6] Revert "chore(generator): centralize mypy configuration and regenerate google-cloud-datastore POC" ↗ googleapis/google-cloud-python
  7. [7] refactor(pubsub): remove EOL Python 3.7/3.8/3.9 false positives and compatibility checks ↗ googleapis/google-cloud-python
  8. [8] chore: update action versions for node24 (#17462) googleapis/google-cloud-python

Quick answers

What shipped in Google on June 18, 2026?
JAX's Pallas GPU kernel interpreter is now thread-efficient, cutting unnecessary jit compilations that were exhausting thread pools. In total, 31 commits, 28 pull requests, and 2 releases landed.
Who contributed to Google on June 18, 2026?
3 developers shipped this update, including Cjkkkk, zhumin8, and chalmerlowe.
What were the notable Google updates?
[Pallas][GPU kernel interpreter] Remove explicit conversions to Jax arrays in host callbacks, Remove stale version guards after JAX v0.10.2 release, and Reenable cuDNN SDPA flex attention with version guard >= 9.13 to avoid NaNs.

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?

We use privacy-friendly analytics (Google Analytics, IP-anonymized) to see which pages help readers. No ads, and we never sell your data. See our Privacy Policy.