RepoJournal
Google

Google

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

Pick a date

  1. Wed 20 may
  2. Thu 21 may
  3. Sat 23 may
  4. Sun 24 may
  5. Mon 25 may
  6. Tue 26 may
  7. Thu 28 may
  8. Fri 29 may
  9. Sat 30 may
  10. Sun 31 may
  11. Mon 1 jun
  12. Mon 8 jun
  13. Tue 9 jun
  14. Wed 10 jun
  15. Thu 11 jun
  16. Fri 12 jun
  17. Sat 13 jun
  18. Sun 14 jun
  19. Mon 15 jun
  20. Tue 16 jun
  21. Wed 17 jun
  22. Thu 18 jun
  23. Fri 19 jun
  24. Sat 27 jun
  25. Sun 28 jun
  26. Mon 29 jun
  27. Tue 30 jun
  28. Wed 1 jul
  29. Thu 2 jul
  30. Today 3 jul

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.

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

FAQ

What changed 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.
What should Google teams do about it?
Test JAX 0.10.2+ with GPU kernels in your Pallas workloads to verify thread efficiency gains • Evaluate python-genai v2.9.0-rc1 for new Gemini model support before stable release • Audit google-cloud-pubsub dependencies if pinned to Python < 3.10 and plan migration
Which Google repositories shipped on June 18, 2026?
google/jax, googleapis/python-genai, googleapis/google-cloud-python

Related across the cluster

For your repos

The showcase is a teaser.
Your wire is the product.

Same engine. Different stack. Below: what changes when the wire is yours.

Showcase wire

  • •14 famous open source orgs
  • •One wire per day
  • •Public, generic
  • •Read on the web, when you remember

Your wire

  • →Up to 1,500 of your repos - orgs, deps, vendors
  • →Morning and evening briefs
  • →Action items routed to your team
  • →Slack delivery, email, breaking-news CVE alerts

Want a hands-on demo first? Ask a current user for an invite link.