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

  1. Sat 27 jun
  2. Sun 28 jun
  3. Mon 29 jun
  4. Tue 30 jun
  5. Wed 1 jul
  6. Thu 2 jul
  7. Fri 3 jul
  8. Sat 4 jul
  9. Sun 5 jul
  10. Mon 6 jul
  11. Tue 7 jul
  12. Wed 8 jul
  13. Thu 9 jul
  14. Fri 10 jul
  15. Sat 11 jul
  16. Sun 12 jul
  17. Mon 13 jul
  18. Tue 14 jul
  19. Wed 15 jul
  20. Thu 16 jul
  21. Sat 18 jul
  22. Sun 19 jul
  23. Mon 20 jul
  24. Tue 21 jul
  25. Wed 22 jul
  26. Thu 23 jul
  27. Fri 24 jul
  28. Sat 25 jul
  29. Sun 26 jul
  30. Mon 27 jul
Topics: AI / ML Python Full archive →

The Wire · Showcase

JAX COMPILER PIPELINE TIGHTENS, GENAI SDK FIXES EVENT LOOP CHAOS

By RepoJournal · Filed · About Google

copybara-service[bot] codyoss TrevorBergeron
3 people shipped this
copybara-service[bot] @copybara-service[bot] 3 cited codyoss @codyoss 1 cited TrevorBergeron @TrevorBergeron 1 cited

JAX's Pallas mosaic backend shipped critical test coverage restoration and parameter naming improvements that make IR debugging possible, while the GenAI SDK patched a nasty event loop mismatch that breaks multi-threaded async environments.

JAX absorbed a revert [1] alongside three substantive compiler improvements across the Pallas mosaic stack. The most impactful: parameter names now attach directly to FuncOp IR [2], meaning debug output shows `o_hbm_ref` instead of `arg1` - a quality-of-life win that compounds across every downstream debugging session. Test coverage got restored [3] for debug_print with complex formatting, catching a gap that only exposed itself in edge cases with trailing text. JAX also fixed a sort_jvp sharding bug [4] and shipped hijax compatibility fixes [5] to ensure high/low flag propagation works correctly in dynamic traces. Over in GenAI SDK, the big move is the event loop mismatch fix [6] [7] - a multi-threaded async auth and session bug that would silently corrupt state when different threads created their own event loops. This one's quiet but dangerous; it hits anyone running GenAI in concurrent contexts. Google Cloud Python cleaned house by removing the legacylibrarian generator [8] [9], the last vestige of an old release system now that release-please owns everything. BigFrames added local UDF execution support [10] [11], expanding compute options for DataFrame operations.

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] Reverts faf677af0abec7ea25c7a33831fcabf30a795796 ↗ google/jax
  2. [2] [pallas:mosaic] Attach Pallas-level parameter names to the `FuncOp` parameters ↗ google/jax
  3. [3] [pallas:mosaic] Re-added a few `pl.debug_print` test cases google/jax
  4. [4] Fix a bug where sort_jvp was not creating the iota primal with the correct sharding google/jax
  5. [5] hijax fixes (v2) google/jax
  6. [6] fix: event loop mismatch in GenAI SDK ↗ googleapis/python-genai
  7. [7] fix: event loop mismatch in GenAI SDK googleapis/python-genai
  8. [8] chore: remove legacylibrarian generator ↗ googleapis/google-cloud-python
  9. [9] chore: remove legacylibrarian generator (#17590) googleapis/google-cloud-python
  10. [10] feat(bigframes): Enable local udf execution ↗ googleapis/google-cloud-python
  11. [11] feat(bigframes): Enable local udf execution (#17588) googleapis/google-cloud-python

Quick answers

What shipped in Google on June 30, 2026?
JAX's Pallas mosaic backend shipped critical test coverage restoration and parameter naming improvements that make IR debugging possible, while the GenAI SDK patched a nasty event loop mismatch that breaks multi-threaded async environments. In total, 18 commits and 16 pull requests landed.
Who contributed to Google on June 30, 2026?
3 developers shipped this update, including copybara-service[bot], codyoss, and TrevorBergeron.
What were the notable Google updates?
Reverts faf677af0abec7ea25c7a33831fcabf30a795796, [pallas:mosaic] Attach Pallas-level parameter names to the `FuncOp` parameters, and [pallas:mosaic] Re-added a few `pl.debug_print` test cases.

More from Google

Google Google

JAX SYNCS XLA COMPILER STACK

JAX updated its XLA dependency to pull in the latest compiler improvements and optimizations.

jax authors
ai 1 shipped 1-min read
Google Google

JAX OVERHAULS AUTODIFF INTERNALS WITH CUSTOM_VJP3, EXPANDS MESH API

JAX shipped a major redesign of custom differentiation rules that simplifies how you define vector-Jacobian products, while simultaneously opening up mesh operations to the public API.

mattjj copybara-service kodlan
ai 24 shipped 1-min read
Google Google

JAX FIXES REMAT3 CONSTANT HANDLING CRASH, GOOGLE CLOUD PYTHON CENTRALIZES REST TRANSCODING

Matthew Johnson patched a critical remat3 regression where closed-over constants were silently dropped, causing immediate crashes in production rematerialization pipelines [ref:1].

Matthew Johnson Sergei Lebedev copybara-service Heba Alazzeh +2
ai 50 shipped 1-min read
Google Google

JAX TIGHTENS XLA OPTIMIZATION CONTROLS, GOOGLE CLOUD PYTHON SETS UP FEATURE GATES

JAX makes XLA optimization levels human-readable while Google Cloud Python client libraries prepare a unified feature gating system for observability.

copybara-service Yash Katariya hebaalazzeh chalmerlowe +1
ai 54 shipped 1-min read

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

AI Agents AI Agents

CLINE SHIPS FIVE CRITICAL AGENT FIXES AFTER END-TO-END TESTING BLITZ

Saoudrizwan just landed a cascade of fixes that prevent the agent loop from running wild, stop file reads from resolving to the wrong directory, and plug the compaction pipeline.

langchain-oss-model-profiles langchain-oss-model-profiles[bot] saoudrizwan
ai-agents 16 shipped 2-min read
Local LLMs Local LLMs

LLAMA.CPP SHIPS MINIMAX-M3 SPARSE ATTENTION, VLLM UNLOCKS HETEROGENEOUS PREFILL-DECODE

The inference stack gets a new model architecture and a fundamental shift in how prefill and decode can run on different hardware configurations.

github-actions[bot] dhiltgen timkhronos Pascal +10
llms 125 shipped 1-min read
CachyOS @CachyOS

GLIBC 2.43 REBUILD SHIPS, INSTALLER JUMPS TO C++26

CachyOS pushed glibc 2.43 through the pipeline while the CLI installer upgraded its C++ standard to C++26, signaling a major refresh across core infrastructure.

Piotr Gorski github-actions Vladislav Nepogodin ptr1337 +7
infra 97 shipped 1-min read
Elixir & Phoenix Elixir & Phoenix

ECTO FIXES SUBQUERY PLANNING IN JOINS, ELIXIR PATCHES IEX PARSING

Ecto's query planner now handles subqueries correctly in interpolated join filters, while Elixir fixes a syntax error that broke 'not in' as a continuation operator in IEx.

Benjamin Milde Alexander Gubarev lukaszsamson makoto-developer
elixir 10 shipped 1-min read
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.